[JAVA] 백준 5567번 : 결혼식

 

---

 

• 코드

1. 처음 if(arr[1][i]==1) 조건문에서 상근이의 친구를 찾는다.

2. 친구를 찾은 다음 상근이의 친구를 찾기위해서 if 문안에서 for문을 이용해서 찾는다.

결과는 맞았습니다!!

 

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

public class Main {
	
	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
        int N = Integer.parseInt(br.readLine());
        int M = Integer.parseInt(br.readLine());
        int [][]friend = new int[N+1][N+1];
		boolean []check = new boolean[N+1];
		int count = 0;

		for(int i=0; i<M; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			int a = Integer.parseInt(st.nextToken());
			int b = Integer.parseInt(st.nextToken());
			friend[a][b] = friend[b][a] = 1;	
		}

		for(int i=2; i<=N; i++) {
			if(friend[1][i] == 1) {	 
				if(!check[i]) {		 
					count ++;		 
					check[i] = true; 
				}

				
				for(int j=2; j<=N; j++) {
					if(friend[i][j] == 1 && !check[j]) {
						count ++;			
						check[j] = true;	
					}
				}
			}
		}

		System.out.println(count);
	}
}