[백준] 1058번 친구

문제

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

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

입력

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

출력

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

예제 입력 1

3

NYY

YNY

YYN

예제 출력 1

2

 

소스코드

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define INF 1e9

int main() {
    int n, cnt, maxNum=0;
    int num[51][51]={0,};
    char c[52][52];
    
    cin>>n;
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=n;j++) {
            cin>>c[i][j];
            if(i==j) num[i][j]=0;	// 자기 자신인 경우 0을 저장한다.
            else if(c[i][j]=='Y') num[i][j]=1;	// 입력값이 Y인 경우, 1을 저장한다.
            else num[i][j]=INF;	// 입력값이 N인 경우, INF를 저장한다.
        }
    }
    // 플로이드-와샬 알고리즘
    for(int k=1;k<=n;k++) {
        for(int i=1;i<=n;i++) {
            for(int j=1;j<=n;j++) {
                if(i==j || j==k || k==i) continue;	// 자기 자신인 경우를 배제한다.
                else if(num[i][j]>num[i][k]+num[k][j])	// INF인 경우, 2-친구의 수를 계산하여 저장한다.
                    num[i][j]=num[i][k]+num[k][j];
            }
        }
    }
    
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=n;j++) {
            if(i==j) continue;	// 자기 자신인 경우를 배제한다.
            else {	// 2-친구인 경우, cnt를 증가시킨다.
                if(num[i][j]<=2) cnt++;
            }
        }
        maxNum=max(maxNum, cnt);	// cnt 값이 최대인 경우를 찾는다.
        cnt=0;
    }
    cout<<maxNum<<endl;
    
    return 0;
}