[BaekJoon] 5567번 - 결혼식 [Java][C++]

문제 링크

---

1. 문제 풀이

초대할 사람은 상근이의 친구과 친구의 친구까지다. 상근이를 기준으로 거리가 2 이내인 모든 사람의 수를 구하면 되므로 최단 거리 BFS를 활용해서 개수를 세주었다.

---

2. 코드

1. 풀이 [Java]

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int n = Integer.parseInt(br.readLine());
        int m = Integer.parseInt(br.readLine());

        List<List<Integer>> adj = new ArrayList<>();
        for (int i = 0; i <= n; i++) {
            adj.add(new ArrayList<>());
        }

        for (int i = 0; i < m; i++) {
            st = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());
            adj.get(a).add(b);
            adj.get(b).add(a);
        }

        int cnt = bfs(n, adj);
        System.out.println(cnt);
    }

    static int bfs(int n, List<List<Integer>> adj) {
        Queue<Integer> q = new ArrayDeque<>();
        q.offer(1);

        boolean[] visited = new boolean[1 + n];
        visited[1] = true;

        int dist = 0;
        int cnt = 0;

        while (!q.isEmpty()) {
            int size = q.size();

            while (size-- > 0) {
                int node = q.poll();

                for (int next : adj.get(node)) {
                    if (visited[next]) continue;

                    q.offer(next);
                    visited[next] = true;
                    cnt++;
                }
            }

            dist++;
            if (dist == 2) break;
        }

        return cnt;
    }
}

2. 풀이 [C++]

#include <bits/stdc++.h>
using namespace std;

int n, m;
vector<vector<int>> adj(1 + 500);
bool visited[1 + 500];

int bfs() {
    queue<int> q;
    q.push(1);

    visited[1] = true;

    int dist = 0;
    int cnt = 0;

    while (!q.empty()) {
        int sz = q.size();

        while (sz--) {
            int node = q.front();
            q.pop();

            for (int next : adj[node]) {
                if (visited[next]) continue;

                q.push(next);
                visited[next] = true;
                cnt++;
            }
        }

        dist++;
        if (dist == 2) break;
    }

    return cnt;
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n >> m;

    for (int i = 0; i < m; i++) {
        int a, b;
        cin >> a >> b;
        adj[a].push_back(b);
        adj[b].push_back(a);
    }

    int cnt = bfs();
    cout << cnt;
}

---