[백준 Java] 1058번 친구(브루트포스 , 그래프 이론)

[Silver II] 친구 - 1058

문제 링크

성능 요약

메모리: 11984 KB, 시간: 84 ms

분류

브루트포스 알고리즘, 플로이드–워셜, 그래프 이론, 그래프 탐색

문제 설명

지민이는 세계에서 가장 유명한 사람이 누구인지 궁금해졌다. 가장 유명한 사람을 구하는 방법은 각 사람의 2-친구를 구하면 된다. 어떤 사람 A가 또다른 사람 B의 2-친구가 되기 위해선, 두 사람이 친구이거나, A와 친구이고, B와 친구인 C가 존재해야 된다. 여기서 가장 유명한 사람은 2-친구의 수가 가장 많은 사람이다. 가장 유명한 사람의 2-친구의 수를 출력하는 프로그램을 작성하시오.

A와 B가 친구면, B와 A도 친구이고, A와 A는 친구가 아니다.

입력

첫째 줄에 사람의 수 N이 주어진다. N은 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에 각 사람이 친구이면 Y, 아니면 N이 주어진다.

출력

첫째 줄에 가장 유명한 사람의 2-친구의 수를 출력한다.

💡 나의 접근

사람을 노드로 관계를 간선으로 생각하고 그래프를 그려보았더니 문제에서 요구하는 2-친구 관계라는 의미가 한다리 건너서 아는 사람까지 친구로 본다는 말이었다는 걸 이해했다.

 

그래서 모든 노드들을 한번씩 탐색하면서 그 노드를 중심으로

(바로 아는 친구 + 한다리 건너서 아는 친구의 수)

가 가장 큰 값을 출력하면 되겠구나를 생각했다.

 

(처음에는 모든 노드 중 가장 간선을 많이 가지는 노드가 가장 친구를 많이 가지겠구나 생각했는데 아닌 경우도 있었다.

예제 2와 같은 경우는 가장 많은 간선을 가지는 노드는 여러개 였고 그 여러개 노드 중 단 하나의 노드만 4명의 친구를 가질 수 있었기 때문이다. )

 

depth라는 변수를 두어서 한다리 건너 아는 친구 까지만 카운팅 되도록 하였으며  자료구조 큐를 사용해서 노드를 건너서 탐색할 수 있게 구현해 보았다. 

 

플로이드 워셜로도 풀 수 있다는데 .. 기억이 안나니까 특징만 정리해두자 

 

플로이드 워셜 - 인접행렬을 이용하여 각 노드간의 최소 비용을 계산하는 알고리즘

특징 : 음수 간선이 존재하고 음수 사이클이 존재하지 않는 그래프에서 사용 가능

 

 

 

💡 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;

public class Main {
	static int N;
	static char[][] friend;
	static boolean[] visited;
	static int maxCount = 0;

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		N = Integer.parseInt(br.readLine());

		friend = new char[N][N];
		for (int i = 0; i < N; i++) {
			friend[i] = br.readLine().toCharArray();
		}
		
		//--------입력

		//완탐
		//각 사람마다 2-친구 수가 몇명인지 알아내서 가장 큰 수 출력
		for (int i = 0; i < N; i++) {
			
			visited = new boolean[N];
			visited[i] = true;

			Queue<Integer> q = new LinkedList<>();
			q.add(i);
			
			int depth = 0;
			int count = 0;

			while (depth < 2) { //한 친구 건너서 알게되는 사람까지만 count하기 때문에
				
				int size = q.size();
				for (int j = 0; j < size; j++) {
					
					int n = q.poll();
					
					for (int k = 0; k < N; k++) {
						
						if (friend[n][k] == 'Y' && !visited[k]) {
							q.add(k);
							count++;
							visited[k] = true;
						}
					}
				}
				depth++;
			}
			if(maxCount < count) {
				maxCount = count;
			}
		}

		System.out.println(maxCount);

	}

}