백준 1707 이분 그래프

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

1707번: 이분 그래프

문제

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

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

입력

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

출력

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

예제 입력 1 

2

3 2

1 3

2 3

4 4

1 2

2 3

3 4

4 2

예제 출력 1 

YES NO

---

[풀이 과정]

1. 각 테스트케이스마다 벡터, 배열들을 사용해야하므로 memset을 이용해 초기화한다.

2. 주어진 정점과 간선을 graph 벡터에 표시한다.

3. 1~n까지 정점 중 방문하지 않은 정점의 그룹을 체크하기 위해 check_group 함수를 실행한다.

4. check_group(int node)

- 현재 정점 방문 체크 후 그룹 번호는 1로 지정한다.

- 큐에 현재 정점을 넣는다.

- 큐에 값이 없을 때까지 현재 정점에서 이동 가능한 정점을 탐색한다.

- 다음 정점이 이미 그룹 번호가 정해져있고 그게 인접한 현재 정점과 같은 번호이면 이분 그래프가 될 수 없으므로 flag를 true로 바꾼 후 함수를 종료한다. (이분 그래프가 되려면 인접한 정점의 그룹 번호는 달라야한다.)

- 다음 정점이 아직 그룹 번호가 정해지지 않았다면 현재 그룹 번호와 다른 번호를 지정 후 큐에 넣는다.

- 모든 경로를 탐색 후 함수를 종료한다.

5. flag가 true이면 NO, false이면 YES를 출력한다.

 

처음 풀었을 때는 3번을 해주지않고 바로 check(1)을 해서 틀렸었다. 문제에는 모든 정점이 연결되어있다는 조건이 없기때문에 1~n번 정점을 모두 방문했는지 체크해야한다.

 

---

[소스 코드]

#include <stdio.h>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
const int MAX = 20005;

int T, n, m, num, group[MAX];
vector <int> graph[MAX];
bool visited[MAX], flag;

void check_group(int node) {
	group[node] = 1;
	visited[node] = true;
	queue <int> q;
	q.push(node);

	while (!q.empty()) {
		int cur = q.front();
		q.pop();

		for (int i = 0; i < graph[cur].size(); i++) {
			int next = graph[cur][i];

			if (group[next] > 0 && group[next] == group[cur]) {
				flag = true;
				return;
			}

			if (group[cur] == 1) num = 2;
			else num = 1;

			if (!visited[next] && group[next] == 0) {
				visited[next] = true;
				group[next] = num;
				q.push(next);
			}
		}
	}
}
int main() {
	scanf("%d", &T);

	for (int t = 1; t <= T; t++) {
		scanf("%d %d", &n, &m);

		memset(graph, 0, sizeof(graph));
		memset(visited, 0, sizeof(visited));
		memset(group, 0, sizeof(group));
		flag = false;

		for (int i = 0; i < m; i++) {
			int a, b;
			scanf("%d %d", &a, &b);
			graph[a].push_back(b);
			graph[b].push_back(a);
		}
		
		
		for (int i = 1; i <= n; i++) {
			if (!visited[i]) check_group(i);
		}

		if (flag) printf("NO\n");
		else printf("YES\n");
	}
	return 0;
}

https://github.com/chojaehyo/CodingTest/blob/main/%5BBOJ%5D%EC%9D%B4%EB%B6%84%20%EA%B7%B8%EB%9E%98%ED%94%84

GitHub - chojaehyo/CodingTest