boj5567 결혼식

링크

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

문제

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

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

입력

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

출력

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

나의 풀이

• dfs 로 친구관계의 depth 가 2 가 되면 return 하도록 계산하였다.
• 핵심 코드

  private static void dfs(int depth,int now){
          if(depth==2){
              return;
          }
          for (int i = 1; i <= N; i++) {
              if(!relation[now][i]) continue;
              if(!checked[i]){
                  cnt+=1;
              }
              checked[i] = true;
              dfs(depth+1,i);
          }
  }

• 전체 코드

  package Baekjoon.java.silver;
  
  import java.io.BufferedReader;
  import java.io.InputStreamReader;
  import java.util.ArrayList;
  import java.util.LinkedList;
  import java.util.Queue;
  import java.util.StringTokenizer;
  
  public class boj5567 {
      static int N,M,cnt;
      static boolean[] checked;
      static boolean[][] relation;
      public static void main(String[] args) throws Exception{
          BufferedReader br = new BufferedReader (new InputStreamReader(System.in));
          N = Integer.parseInt(br.readLine());
          M = Integer.parseInt(br.readLine());
          cnt = 0;
          checked = new boolean[N+1];
          relation = new boolean[N+1][N+1];
          StringTokenizer st;
          for (int i = 0; i < M; i++) {
              st = new StringTokenizer(br.readLine());
              int a = Integer.parseInt(st.nextToken());
              int b = Integer.parseInt(st.nextToken());
  
              relation[a][b]= true;
              relation[b][a] = true;
          }
          checked[1] = true;
          dfs(0,1);
          System.out.println(cnt);
      }
      private static void dfs(int depth,int now){
          if(depth==2){
              return;
          }
          for (int i = 1; i <= N; i++) {
              if(!relation[now][i]) continue;
              if(!checked[i]){
                  cnt+=1;
              }
              checked[i] = true;
              dfs(depth+1,i);
          }
      }
  }

결과