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;
}