이분 그래프 성공

시간 제한메모리 제한제출정답맞힌 사람정답 비율

2 초

256 MB

55255

14280

8429

23.257%

문제

그래프의 정점의 집합을 둘로 분할하여, 각 집합에 속한 정점끼리는 서로 인접하지 않도록 분할할 수 있을 때, 그러한 그래프를 특별히 이분 그래프 (Bipartite Graph) 라 부른다.

그래프가 입력으로 주어졌을 때, 이 그래프가 이분 그래프인지 아닌지 판별하는 프로그램을 작성하시오.

입력

입력은 여러 개의 테스트 케이스로 구성되어 있는데, 첫째 줄에 테스트 케이스의 개수 K가 주어진다. 각 테스트 케이스의 첫째 줄에는 그래프의 정점의 개수 V와 간선의 개수 E가 빈 칸을 사이에 두고 순서대로 주어진다. 각 정점에는 1부터 V까지 차례로 번호가 붙어 있다. 이어서 둘째 줄부터 E개의 줄에 걸쳐 간선에 대한 정보가 주어지는데, 각 줄에 인접한 두 정점의 번호 u, v (u ≠ v)가 빈 칸을 사이에 두고 주어진다.

출력

K개의 줄에 걸쳐 입력으로 주어진 그래프가 이분 그래프이면 YES, 아니면 NO를 순서대로 출력한다.

제한

2 ≤ K ≤ 5
1 ≤ V ≤ 20,000
1 ≤ E ≤ 200,000

예제 입력 1 복사

예제 출력 1 복사

YES
NO

DFS를 이용한 풀이

: 일단 이분 그래프이기 때문에 정점을 두 팀으로 나눈다고 생각하면 편하다.

그래서 DFS를 수행하면서 만약 1-2, 2-4 이렇게 인접해있다고 하자. 1번이 A팀이라면 1번과 인접해있는 2번은 B팀으로 보내고, 2번과 인접해있는 4번은 다시 A팀으로 보내면 된다.

틀린 이유

1. 메모리 초과

: 처음에 인접 행렬을 이용해 메모리 초과가 일어났다.

2. result를 초기화 하지 않음

: 테스트 케이스만큼 반복하기 때문에 새로운 테스트를 할 때마다 result를 초기화 시켜줘야 했는데 이것 때문에 계속 틀렸다는 결과가 나왔다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.StringTokenizer;

public class Main {
    static int[] visited;
    static ArrayList<Integer>[] adjacent;
    static int v;
    static boolean result;
    public static void main(String args[]) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        int k = Integer.parseInt(br.readLine());
        for (int i = 0; i < k; i++) {
            st = new StringTokenizer(br.readLine());
            result = false;
            v = Integer.parseInt(st.nextToken());
            visited = new int[v+1];
            adjacent = new ArrayList[v+1];
            for (int j = 1; j <= v; j++) {
                adjacent[j] = new ArrayList<>();
            }
            int e = Integer.parseInt(st.nextToken());
            for (int j = 0; j < e; j++) {
                st = new StringTokenizer(br.readLine());
                int a = Integer.parseInt(st.nextToken());
                int b = Integer.parseInt(st.nextToken());
                adjacent[a].add(b);
                adjacent[b].add(a);
            }

            for (int j = 1; j <= v; j++) {
                if (visited[j] == 0) {
                    dfs(j, 1);
                }
            }

            if (result) {
                System.out.println("NO");
            }
            else {
                System.out.println("YES");
            }
        }

    }

    static void dfs(int j, int col) {
        visited[j] = col;
        if (result) //만약 result가 이미 true라면 더 수행할 필요없이 종료 
            return;
        for (Integer i : adjacent[j]) {
            if (visited[i] == visited[j]) {
                result = true;
                return;
            }
            if (visited[i] == 0) {
                dfs(i, col*-1);
            }
        }
    }
}