演算法思路:帶權並查集。
但是要注意“權值”的選取,下面的代碼採取的方案是:初始的時候儲存每個節點到父節點的距離d[i], 當進行一次findset的時候同時壓縮路徑和依次更新路徑上的每個結點的d[i], 使得此時的的d[i]為該節點到根節點的距離,執行findset 後cout<<d[i]就是題目的結果啦。(本菜鳥個人認為這種思想比較經典,有代表性,有助於類似的題目啟發思維)
代碼來源:劉汝佳 - 資訊學競賽入門經典(強烈推薦,難得的好書)
//模板開始#include <string> #include <vector> #include <algorithm> #include <iostream> #include <sstream> #include <fstream> #include <map> #include <set> #include <cstdio> #include <cmath> #include <cstdlib> #include <ctime>#include<iomanip>#define SZ(x) (int(x.size()))using namespace std;int toInt(string s){istringstream sin(s); int t; sin>>t; return t;}template<class T> string toString(T x){ostringstream sout; sout<<x; return sout.str();}typedef long long int64;int64 toInt64(string s){istringstream sin(s); int64 t; sin>>t;return t;}template<class T> T gcd(T a, T b){ if(a<0) return gcd(-a, b);if(b<0) return gcd(a, -b);return (b == 0)? a : gcd(b, a % b);}//模板結束(通用部分)#define ifs cinconst int maxn = 20000 + 10;int pa[maxn], d[maxn];int findset(int x){if(pa[x] != x){int root = findset(pa[x]);d[x] += d[pa[x]];return pa[x] = root;}else{return x;}}//LA 3027 Corporative Networkint main(){//ifstream ifs("shuju.txt", ios::in);int T;ifs>>T;while(T--){int n, u, v;char cmd[9];ifs>>n;for(int i = 1; i <= n; i++)//初始化,每個節點單獨是一顆樹{pa[i] = i;d[i] = 0;}while(ifs>>cmd && cmd[0] != 'O'){if(cmd[0] == 'E'){ifs>>u;findset(u);cout<<d[u]<<endl;}if(cmd[0] == 'I'){ifs>>u>>v;pa[u] = v;d[u] = abs(u - v) % 1000;}}}return 0;}