天梯赛 L2-023. 图着色问题

题解:用dfs遍历图的每条边就好，这里注意要求颜色的个数为k

#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
#include <map>
#include <vector>
#include <set>
using namespace std;
const int maxn=;
int v,e,k;
vector<int> edge[maxn];
int val[maxn];
int vis[maxn];
int flag;
void dfs(int now,int fa)
{
    if(flag==) return ;
    int len=edge[now].size();
    for(int i=;i<len;i++)
    {
        int next=edge[now][i];
        if(val[now] == val[next])
        {
            flag=;
            return;
        }
        if(next != fa &&  vis[next] ==  )
        {
            vis[next]=;
            dfs(next,now);
        }
    }
}
int main()
{
    scanf("%d %d %d",&v,&e,&k);
    for(int i=;i<=e;i++)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        edge[b].push_back(a);
        edge[a].push_back(b);
    }
    int q;
    cin>>q;
    while(q--)
    {
        set<int> st;
        memset(vis,,sizeof(vis));
        flag=;
        for(int i=;i<=v;i++)
        {
            scanf("%d",&val[i]);
            st.insert(val[i]);
        }
        if(st.size() != k) cout<<"No"<<endl;
        else
        {
            for(int i=;i<=v;i++)
            {
                if(flag==) break;
                if(!vis[i])
                {
                    vis[i]=;
                    dfs(i,i);
                }
            }
            if(flag==) cout<<"No"<<endl;
            else cout<<"Yes"<<endl;
        }
    }
    return ;
}