Cộng đồng Hỏi & Đáp của Inflearn

안녕하세요 강의 잘 듣고 있습니다.

해당 문제 테스트 케이스에 오류가 있는 것 같아 글 남깁니다!

연결된 네트워크 그룹이 여러 개일 때를 고려하지 않고 문제를 풀었는데 정답 처리 되었습니다..!

확인 부탁드립니다

감사합니다!

import java.util.*;

class Solution {
    List<List<Integer>> graph;
    
    public boolean solution(int[][] friends) {
        graph = new ArrayList<>();
        for (int i = 0; i < friends.length; i++) {
            graph.add(new ArrayList<>());
        }
        
        for (int i = 0; i < friends.length; i++) {
            for (int j = 0; j < friends[i].length; j++) {
                graph.get(i).add(friends[i][j]);
                graph.get(friends[i][j]).add(i);
        
            }
        }

        return bfs(0);
    }

    private boolean bfs(int start) {
        Queue<Integer> queue = new ArrayDeque<>();
        queue.offer(start);
        
        int[] team = new int[graph.size()];
        Arrays.fill(team, -1);
        team[start] = 0; // 시작 팀

        while(!queue.isEmpty()) {
            int cur = queue.poll();

            for(int next : graph.get(cur)) {
                if (team[next] == -1) { // 아직 미방문한 노드
                    team[next] = 1 - team[cur]; // 반대팀 배정
                    queue.offer(next);
                    
                } else if (team[next] == team[cur]) { // 방문한 노드가 같은 팀이면 false 반환
                    return false;
                }
            }
        }
        return true;
        
    }
}