PAT甲级 1154 Vertex Coloring (25 分)

题意

检查一幅图是否所有的相邻两点颜色都不一样。

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1e4 + 10;
typedef long long ll;

struct edge{
	int u, v;
}arr[N];
int col[N];

int main(void)
{
	int i, j, temp;
	int n, m;
	scanf("%d%d", &n, &m);
	for (i = 1; i <= m; i++)
	{
		scanf("%d%d", &arr[i].u, &arr[i].v);
	}
	
	int q;
	scanf("%d", &q);
	while (q--)
	{
		set<int> se;
		for (i = 0; i < n; i++)
		{
			scanf("%d", col+i);
			se.insert(col[i]);
		}
		for (i = 1; i <= m; i++)
		{
			if (col[arr[i].u] == col[arr[i].v])
			{
				printf("No\n");
				break;
			}
		}
		if (i == m+1)
			printf("%d-coloring\n", se.size());
	}
}