程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> 關於C++ >> uvaliva3027(並查集)

uvaliva3027(並查集)

編輯:關於C++

題意:

如果是Iuv則是把u的父節點設置為v;並且u到v的距離為|u-v| % 1000;

如果Eu 則輸出u到根的距離;

O結束;

 

思路:

在合並階段就是普通的並查集,但還需要算一個距離:

但每次查詢時,就應該把距離累加起來,並記錄下來:

 

AC代碼:

 

#include
#include
#include
using namespace std;

const int N = 20000 + 5;
int p[N];
int dis[N];
int n;

void init(int n) {
	for(int i = 0; i <= n; i++) {
		p[i] = i;
		dis[i] = 0;
	}
	return ;
}

int find_parent(int x) {
	if(x != p[x]) {
		int r = find_parent(p[x]);
		dis[x] += dis[p[x]];
		return p[x] = r;
	}
	return x;
}
void Union(int u, int v) {
	p[u] = v;
	dis[u] = abs(u - v);
	dis[u] %= 1000;
}
int main() {
	int t;
	scanf("%d",&t);
	while(t--) {
		scanf("%d",&n);
		init(n);
		char c;
		while(1) {
			getchar();
			c = getchar();
			if(c == 'I') {
				int a,b;
				scanf("%d%d",&a,&b);
				Union(a,b);
			}
			else if(c == 'E') {
				int a;
				scanf("%d",&a);
				find_parent(a);
				printf("%d\n",dis[a]);
			}
			else
				break;
		}
	}
}


 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved