#백준_5567 결혼식 - Java 자바

#유형 : 그래프 탐색, BFS

# 난이도 : 실버 1

# 전형적인 BFS문제에 조건 하나 추가해준 문제. 난이도에 맞지 않게 매우 쉬웠다고 생각함. 큐에 Point형으로 x,y 값을 넣는데 친구의 친구까지만 허용되므로 2보다 큰 값인 경우에는 허용되지 않는다.

package bj;
 
import java.awt.Point;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
 
public class p5567 {
    static ArrayList<Integer> arrList = new ArrayList<>();
    static int N,M;
    static int arr[][];
    static boolean visit[];
    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        M = Integer.parseInt(br.readLine());
        arr = new int[N+1][N+1];
        visit = new boolean[N+1];
        for(int m=0; m<M; m++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int u = Integer.parseInt(st.nextToken());
            int v = Integer.parseInt(st.nextToken());
            arr[u][v] = 1;
            arr[v][u] = 1;
        }
        bfs();
        System.out.println(arrList.size());
//        for(int i : arrList)
//            System.out.println(i);
    }
    private static void bfs() {
        // TODO Auto-generated method stub
        Queue<Point> queue = new LinkedList<Point>();
        queue.add(new Point(1,0));
        visit[1] = true;
        
        while(!queue.isEmpty()) {
            Point po = queue.poll();
            
            for(int i=1; i<=N; i++) {
                if(arr[po.x][i] == 1 && !visit[i] && po.y < 2) {
                    arrList.add(i);
                    visit[i] = true;
                    queue.add(new Point(i, po.y+1));
                }
            }
        }
    }
}