c언어 17250번 은하철도 (백준)

#include <stdio.h>

int find(int p[], int x) { //노드가 속한 부분집합의 대표 노드 찾기
	if(x==p[x]) return x;
	return p[x] = find(p, p[x]);
}

void merge(int p[], int x, int y) {//두개의 노드가 속한 부분집합 합치기
	x = find(p, x);
	y = find(p, y);
	if(x!=y) p[x] = y;
}

int same(int p[], int x, int y) {//두 노드의 부분집합은 같은가?
	x = find(p, x);
	y = find(p, y);
	return x==y;
}

int main(void) {
	int n, m;
	scanf("%d %d", &n, &m); //n, m입력
	int p[n+1]; //노드가 속한 부분집합의 대표 노드
	int cnt[n+1]; //노드가 가지는 철도의 개수
	int save[n+1]; //부분집합의 대표가 가지는 철도의 개수
	for(int i=1; i<=n; i++) { //철도의 개수 입력받기
		scanf("%d", &cnt[i]);
		p[i] = i;
		save[i] = cnt[i];
	}
	
	while(m--) {
		int a, b;
		scanf("%d %d", &a, &b);
		
		if(same(p, a, b)) { //같은 집합이면
			printf("%d\n", save[find(p,a)]); // 그 부분집합의 대표가 가지는 철도 개수 출력
		} else {
		//다른 집합이면
		int tmp = save[find(p,a)] + save[find(p,b)]; 각각 노드가 속한 부분집합의 대표가 가지는 철도의 개수들을 합침
		merge(p, a, b); //집합들을 합침
		save[find(p,a)] = tmp; //합쳐진 집합의 대표가 가지는 철도의 개수들을 합침
		printf("%d\n", tmp); //출력
        }
	}
}

 

풀이 : union(합집합) find(찾기)

1. union find 을 위한 함수들을 만듭니다.

2. 배열들을 만듭니다. (노드의 부분집합의 대표, 노드가 가지는 철도의 개수, 부분집합의 대표가 가지는 철도의 전체 개수

3. 은하의 개수 입력받기

4. m만큼 두 은하를 입력 받은 뒤, 그 두 은하가 속한 집합의 철도의 개수를 출력한다.

4.1 두개의 은하가 같은 집단이면, 그 하나의 집단의 철도의 개수를.

4.2 두개의 은하가 다른 집단이면, 두 집단의 철도의 개수를 합친 다음 출력한다 (이 경우에는 save를 두 집단의 철도의 개수로 바꾸어야 한다)

 

https://www.acmicpc.net/problem/17250

17250번: 은하철도