1154 Vertex Coloring (25 分)

题目链接

题意

判断给定的图中，是否每条边的两个顶点颜色不相同？如果是，则判断使用了几种颜色；如果不是，则输出no。

分析

1.图只给了一次，但要判断很多种不同的着色方案，因此要存储边信息
2.将顶点着色编号存两次，一次是编号对应的着色，一次是存入set数组，计算色彩种类（或者用map也行）
3.记录相同颜色的边数

思维误区

1.使用unordered-map存储边，但一个顶点可能和几个顶点有映射关系，导致边信息被遗漏。因此边的存储选用了结构体数组

代码

#include <bits/stdc++.h>
using namespace std;
struct node{
    int e1,e2;
};
int main()
{
    int n,m,k,e1,e2,t;
    scanf("%d %d",&n,&m);
    vector<node> e(m);
    for(int i=0;i<m;i++){
        scanf("%d %d",&e[i].e1,&e[i].e2);
    }
    scanf("%d",&k);
    for(int i=0;i<k;i++){
        vector<int> v(n);
        set<int> s;
        int cnt=0;
        for(int j=0;j<n;j++){
            scanf("%d",&v[j]);
            s.insert(v[j]); //mp[v[j]]++;
        }
        for(int j=0;j<m;j++){
            if(v[e[j].e1]==v[e[j].e2]) cnt++;
        }
        if(cnt!=0) printf("No\n");
        else printf("%d-coloring\n",s.size());
    }
}