構建帶權值邊的並查集,對於每個方程式有兩種情況:
查詢是否聯通,若不聯通,則答案為-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的正方形區域都滿足這樣的性質 左上角的數字 右下角的數字 左下角的數字 右上角的數字.有些格仔已經確定,...