帶權並查集 Leetcode 399 除法求值

2021-10-06 16:19:49 字數 1829 閱讀 8581

構建帶權值邊的並查集,對於每個方程式有兩種情況:

查詢是否聯通,若不聯通,則答案為-1.0

若聯通,則求出其與根相除的結果,根據結果計算方程式。

例如,對於題目示例:

構造並查集:a->b->c,其中a->b的邊的權值為2.0,b->c的邊的權值為3.0

以計算b/a時為例,以root結點為橋梁,計算b/a的值:

首先計算b/root,這裡root為c,則b/c = 3.0

然後計算a/root,這裡root為c,則a/c = (a/b) * (b/c) = 2.0 * 3.0 = 6.0

最後計算方程式結果:b/a = (b/root)/(a/root) = 3.0/6.0 = 0.5

建立map儲存節點的parent,建立map儲存節點除以parent的值

以a/b=2.0 ,b/c=3.0為例:

find函式用於返回節點的根以及節點除以根的值

union函式在合併的同時需對weight進行更新:

parent[roota.first]

=rootb.first;

weight[roota.first]=1

/roota.second*a_div_b*rootb.second;

roota.second由find(a)的返回值得到,即等於a/a的根

a_div_b等於a/b

rootb.second由find(b)的返回值得到,即等於b/b的根

weight[roota.first]的值應等於 a的根/b的根

即weight[roota.first]=1/roota.second * a_div_b * rootb.second

對每個已知方程式的兩個字串進行union操作

對每個未知方程式的兩個字串進行find操作

若兩個字串的根均存在且相同,則返回兩個find函式的商,即(a/root)/(b/root) = a/b即為所求

class

solution

;double result=

1.0;

while

( parent[a]

!=a )

return;}

void

set_union

( string a,string b,

double a_div_b )

vector<

double

>

calcequation

(vector>

& equations, vector<

double

>

& values, vector>

& queries)

if( parent.

find

(b)==parent.

end())

set_union

( a,b,values[i]);

} vector<

double

> ans;

for( i=

0;isize()

;i++

)return ans;}}

;

並查集,帶權並查集

題意 ignatius過生日,客人來到,他想知道他需要準備多少張桌子。然而一張桌子上面只能坐上相互熟悉的人,其中熟悉可定義成為a與b認識,b與c認識,我們就說a,b,c相互熟悉 例如a與b熟悉and b與c熟悉,d與e熟悉,此時至少需要兩張桌子。輸入 t表示樣例個數,n表示朋友個數,朋友從1到n編號...

帶權並查集

食物鏈 time limit 1000ms memory limit 10000k total submissions 71395 accepted 21146 description 動物王國中有三類動物a,b,c,這三類動物的食物鏈構成了有趣的環形。a吃b,b吃c,c吃a。現有n個動物,以1 n...

帶權並查集 then

問題 b 便 時間限制 2 sec 記憶體限制 512 mb 提交 50 解決 14 題目描述 給出乙個r c的棋盤.共有r行c列,r c個格仔.現要在每個格仔都填乙個非負整數.使得任意乙個2 2的正方形區域都滿足這樣的性質 左上角的數字 右下角的數字 左下角的數字 右上角的數字.有些格仔已經確定,...