백준 5567 : 결혼식

문제

상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.

상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다. 

출력

첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.

예제 입력 1 복사

6
5
1 2
1 3
3 4
2 3
4 5

예제 출력 1 복사

3

예제 입력 2 복사

6
5
2 3
3 4
4 5
5 6
2 5

예제 출력 2 복사

0

 

• 풀이

https://hoshi03.tistory.com/277

백준 1389 : 케빈 베이컨의 6단계 법칙

거의 비슷한 문제, bfs로 그래프 탐색

이 문제는 1번에서 시작해서 각 노드까지의 거리를 재고 그게 1(친구)이거나 2(친구의 친구)면 숫자를 세서 리턴하면 된다

import java.util.*;
import java.io.*;

public class Main{

    static int N, M;

    static boolean[] visit;
    static int[] dist;

    static ArrayList<Integer>[] arr;

    static void bfs(int x){
        Queue<Integer> q = new LinkedList<>();
        visit[x] = true;
        dist[x] = 0;
        q.add(x);
        while (!q.isEmpty()){
            x = q.poll();
            for (int y : arr[x]){
                if (visit[y]) continue;
                q.add(y);
                visit[y] = true;
                dist[y] = dist[x]+1;
            }
        }
    }

    public static void main(String[] args) throws IOException {
        Scanner in = new Scanner(System.in);
        N = in.nextInt();
        M = in.nextInt();
        int ans = 0;

        arr = new ArrayList[N+1];
        visit = new boolean[N+1];
        dist = new int[N+1];
        for (int i = 1; i <= N; i++) arr[i] = new ArrayList<>();
        for (int i = 1; i <=M; i++ ){
            int a = in.nextInt();
            int b = in.nextInt();
            arr[a].add(b);
            arr[b].add(a);
        }

        bfs(1);

        for (int i = 2; i <= N; i++) {
            if (dist[i] == 1 || dist[i] == 2) ans++;
        }
        System.out.println(ans);
    }
}