这题感觉和 POJ 1988 Cube Stacking 很像,在路径压缩的同时递归出来的时候跟新distant数组
我发现我一直WA的原因是,命令结束是以字母o结束的,而不是数字0!!
1 //#define LOCAL 2 #include <algorithm> 3 #include <cstdio> 4 using namespace std; 5 6 const int maxn = 20000 + 10; 7 int parent[maxn], distant[maxn]; 8 9 int GetParent(int a) 10 { 11 if(parent[a] == a) return a; 12 int root = GetParent(parent[a]); 13 distant[a] += distant[parent[a]]; 14 return parent[a] = root; 15 } 16 17 int main(void) 18 { 19 #ifdef LOCAL 20 freopen("3027in.txt", "r", stdin); 21 #endif 22 23 int T, n; 24 scanf("%d", &T); 25 while(T--) 26 { 27 scanf("%d", &n); 28 char cmd[9]; 29 for(int i = 1; i <= n; ++i) 30 { 31 parent[i] = i; 32 distant[i] = 0; 33 } 34 while(scanf("%s", cmd) && cmd[0] != 'O') 35 { 36 int a, b; 37 if(cmd[0] == 'E') 38 { 39 scanf("%d", &a); 40 GetParent(a); 41 printf("%d ", distant[a]); 42 } 43 else 44 { 45 scanf("%d%d", &a, &b); 46 parent[a] = b; 47 distant[a] = abs(a - b) % 1000; 48 } 49 } 50 } 51 return 0; 52 }