문제는 다음과 같다. (골드4)
여느 문제와 같이 변수에 대한 제한이 있고, 이를 지켜주지 않으면 런타임 에러가 난다.(이것때매 헤메기도 했음)
사실 그냥 조금 어려웠다...!
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;
vector<int> a[20001];
int color[20001];
void dfs(int node, int c) {
color[node] = c;
for (int i=0; i<a[node].size(); i++) {
int next = a[node][i];
if (color[next] == 0) {
dfs(next, 3-c);
}
}
}
int main() {
int t;
scanf("%d\n",&t);
while (t--) {
int n, m;
scanf("%d %d",&n,&m);
for (int i=1; i<=n; i++) {
a[i].clear();
color[i] = 0;
}
for (int i=0; i<m; i++) {
int u,v;
scanf("%d %d",&u,&v);
a[u].push_back(v);
a[v].push_back(u);
}
for (int i=1; i<=n; i++) {
if (color[i] == 0) {
dfs(i, 1);
}
}
bool ok = true;
for (int i=1; i<=n; i++) {
for (int k=0; k<a[i].size(); k++) {
int j = a[i][k];
if (color[i] == color[j]) {
ok = false;
}
}
}
printf("%s\n",ok ? "YES" : "NO");
}
return 0;
}
코드는 위와 같다. 백준의 알고리즘 강의를 들어서 그나마 풀 수 있었던거같다. 아이디어를 대강은 이해하고 풀이에 들어간 것이기 때문에 구현하는건 어렵지 않았다. 다만 다른곳에서 애를 좀 먹었다 ㅠ;;
1. 배운것
-vector자체에도 크기 제한이 정해져있다..!
위의 코드와 같이 vector<int>a[20001]; 은 벡터와 배열을 섞어 2차원 배열의 형태를 만든것이다. 다만 이와 비슷한 표 현인 vector<vector<int>> a(20001,vector<int>(20001,0)); 을 선언한다면 컴파일이 안되는...(야발 ㅋㅋ)
그러니까, 2차원 vector는 위와 같이 선언하도록 하자. (인접리스트의 방식인가)
++추가로 지역변수에 선언할 경우 제한이 있다고 한다!
'BOJ 문제풀이' 카테고리의 다른 글
#2178 미로탐색 (0) | 2022.03.29 |
---|---|
#1012 유기농 배추 (0) | 2022.03.27 |
#2667 단지번호 붙이기 (0) | 2022.03.27 |
#11724 연결요소의 개수 (0) | 2022.03.27 |
#2606 바이러스 (0) | 2022.03.27 |