PAT A1154 Vertex Coloring (25 分) 图

    题目大意：给出一张图的信息，要求每条边的两个端点颜色不同。给出K种上色序列，判断是否符合条件。

    用邻接表实现，遍历顶点 i 的所有邻接顶点，如果有和 i 的颜色相同的，就不符合条件。颜色的种类数可以用set存储。

AC代码：

#include <vector>
#include <cstdio>
#include <set>
using namespace std;

const int MAXN = 10010;
vector<int> G[MAXN];

int main()
{
    int N, M, K;
    scanf("%d%d", &N, &M);
    for (int i = 0; i < M; ++i)
    {
        int u, v;
        scanf("%d%d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    scanf("%d", &K);
    for (int q = 0; q < K; ++q)
    {
        set<int> st;
        vector<int> color(N);
        for (int i = 0; i < N; ++i)
        {
            scanf("%d", &color[i]);
            st.insert(color[i]);
        }
        bool flag = true;
        for (int i = 0; i < N && flag; ++i)
        {
            for (int j = 0; j < G[i].size() && flag; ++j)
            {
                if(color[i] == color[G[i][j]])
                    flag = false;
            }
        }
        if(flag) printf("%d-coloring\n", (int)st.size());
        else printf("No\n");
    }
    return 0;
}