입력은 여러 개의 테스트 케이스로 구성되어 있는데, 첫째 줄에 테스트 케이스의 개수 K가 주어진다. 각 테스트 케이스의 첫째 줄에는 그래프의 정점의 개수 V와 간선의 개수 E가 빈 칸을 사이에 두고 순서대로 주어진다. 각 정점에는 1부터 V까지 차례로 번호가 붙어 있다. 이어서 둘째 줄부터 E개의 줄에 걸쳐 간선에 대한 정보가 주어지는데, 각 줄에 인접한 두 정점의 번호 u, v (u ≠ v)가 빈 칸을 사이에 두고 주어진다.
K개의 줄에 걸쳐 입력으로 주어진 그래프가 이분 그래프이면 YES, 아니면 NO를 순서대로 출력한다.
간단하게 이분그래프인지 판별하는 문제이다.
푼 것 후에도 인터넷 서치로 또다른 해설과함께 두개의 방식을 알아냈다.
첫번째 방법
#define RED 1
#define BLUE 2
void dfs(int node)
{
if (!visitied[node])
{
visitied[node] = RED;
}
for (auto i : Arr[node])
{
if (!visitied[i])
{
if (visitied[node] ==RED)
{
visitied[i] = BLUE;
}
else if (visitied[node] == BLUE)
{
visitied[i] = RED;
}
dfs(i);
}
else if (visitied[i] == visitied[node])
{
trueth = false;
}
}코드를 입력하세요
첫번째 방법은 dfs와 red,blue로 설정하여
이분 그래프인지 판단하는 방법이다.
재귀 안에서
else if (visitied[i] == visitied[node])
둘이 같은 색이면
trueth = false;
로 판별 변수를 바꿔
메인 함수에서 dfs를 종료하는 방식이다.
다음은 두번째 방법이다.
visitied[node] = true;
for (auto i : Arr[node])
{
if (!visitied[i])
{
//rsult[node]+1%i;
result[i] = (result[node]+1) % 2;
dfs(i);
}
else if (result[i] == result[node])
{
trueth = false;
}
}
이 방법은 red,blue를 설정하지 않고
바로 result[i] = (result[node]+1) % 2를 i에 넣어
i와 node 값이 같은지 확인후 같으면 이분 그래프가 아님으로
바로 종료하는 로직이다.
두 방법 모두 메인은 똑같이 설정하였다.
#define MAX_SIZE 20000+1
vector<int>Arr[MAX_SIZE];
int visitied[MAX_SIZE];
bool trueth;
#define RED 1
#define BLUE 2
int main()
{
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
int k;
cin >> k;
for (int i = 0; i < k; i++)
{
int n, m;
cin >> n >> m;
trueth = true;
for (int j = 0; j < m; j++)
{
int a, b;
cin >> a >> b;
Arr[a].push_back(b);
Arr[b].push_back(a);
}
for (int w = 1; w <= n;w++)
{
if (trueth)
{
dfs(w);
}
else
{
break;
}
}
if (trueth)
{
cout << "YES" << "\n";
}
else
{
cout << "NO" << "\n";
}
for (int r = 0; r <= n; r++)
{
Arr[r].clear();
visitied[r] = 0;
}
}