[알고리즘] Java / 백준 / 사이클게임 / 20040

[알고리즘] Java / 백준 / 사이클게임 / 20040

문제

문제 링크

접근 방식

유니온 파인드를 구현하고 사이클이 만들어지면 바로 해당 번호를 출력한다
유니온 파인드

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main_20040 {

	static int parent[];
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());
		
		int answer = 0;
		
		parent = new int[N];
		
		for(int i=0;i<N;i++) {
			parent[i] = i;
		}
		
		for(int i=0;i<M;i++) {
			st = new StringTokenizer(br.readLine());
			int x = Integer.parseInt(st.nextToken());
			int y = Integer.parseInt(st.nextToken());
			
			if(union(x,y)) continue;
			answer = i+1;
			break;
		}	
		
		System.out.println(answer);
	}
	
	public static boolean union(int x, int y) {
		x = find(x);
		y = find(y);
		if(x != y) {
			parent[y] = x;
			return true;
		}
		return false;
	}
	
	public static int find(int x) {
		if(x == parent[x]) return x;
		
		return parent[x] = find(parent[x]);
	}

}