[백준] 결혼식 - Silver 2

[jusung-c]

문제 링크

https://www.acmicpc.net/problem/5567

문제 조건

• 1~N의 학번을 가진 N명의 학생이 있다.
• 친구 관계 리스트를 가지고 있을 때 결혼식에 친구와 친구의 친구를 초대하기로 했다.
• 최종적으로 초대할 사람의 수를 구하라

---

아이디어

친구와 친구의 친구라는 것은 그래프에서 최소 간선 횟수가 2 이하인 경우를 뜻한다.

1. 친구 관계를 인접 리스트로 표현한다.
2. 상근이(1)를 시작으로 BFS를 돌면서 최소 간선 횟수를 dist에 갱신한다.
3. dist 배열 중 2 이하인 경우를 구한다.

---

구현

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

public class Main {
    static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));;
    static StringTokenizer st;
    static int N, M;
    static ArrayList<Integer>[] adj;
    static int[] dist;

    public static void init() throws IOException {
        N = Integer.parseInt(br.readLine());
        M = Integer.parseInt(br.readLine());

        adj = new ArrayList[N + 1];
        for (int i = 1; i <= N; i++) {
            adj[i] = new ArrayList<>();
        }

        dist = new int[N + 1];
        for (int i = 1; i <= N; i++) {
            dist[i] = -1;
        }

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken());
            int y = Integer.parseInt(st.nextToken());

            adj[x].add(y);
            adj[y].add(x);
        }
    }

    private static void pro() throws IOException {
        bfs(1);

        int sum = 0;
        for (int i = 1; i <= N; i++) {
            if (dist[i] > 0 && dist[i] <= 2) sum++;
        }

        bw.write(sum + " ");
    }

    private static void bfs(int x) {
        Queue<Integer> que = new LinkedList<>();
        que.add(x);
        dist[x] = 0;

        while (!que.isEmpty()) {
            x = que.poll();

            for (int y : adj[x]) {
                if (dist[y] != -1) continue;

                que.add(y);
                dist[y] = dist[x] + 1;
            }
        }
    }

    public static void main(String[] args) throws Exception {
        init();

        pro();

        bw.close();
    }
}