1154 Vertex Coloring

陌陌的pat甲级题解目录

题解

和之前天梯赛出过的图着色问题本质相同。
主要是判断这个图的着色是否正确：即相邻节点不能有相同的颜色。
判读这个可直接遍历每个点，check当前点的所有临界点。
如果成功，输出使用的颜色数量，用set存即可。

#include <iostream>
#include <vector>
#include <set>
using namespace std;
vector<vector<int>> v;
int N, M;
bool judge(vector<int> x)
{
    for (int i = 0; i < N; i++)
    {
        for (auto &e : v[i])
        {
            if (x[e] == x[i])
                return 0;
        }
    }
    return 1;
}
int main()
{
    cin >> N >> M;
    v.resize(N + 5);
    while (M--)
    {
        int x, y;
        cin >> x >> y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    cin >> M;
    while (M--)
    {
        set<int> ret;
        vector<int> x(N + 5);
        for (int i = 0; i < N; i++)
            cin >> x[i], ret.insert(x[i]);
        if (judge(x))
        {
            cout << ret.size() << "-coloring" << endl;
        }
        else
            cout << "No\n";
    }
    return 0;
}