본문 바로가기

BOJ 문제풀이

#1707 이분 그래프

문제는 다음과 같다. (골드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