PTA L2-023 图着色问题C++

 由题意可知，要求用k种颜色将图的每个顶点染色，相邻点不能有相同的颜色

注意：k种颜色要用完

首先，用邻接表把图存储下来，然后对于每一种方案，判断合不合理，写一个函数进行判断，如果有相邻点是相同的颜色，则不合理，颜色不是k种也不合理

int judge(int cnt){//判断一个方案是否合理，cnt为使用的颜色种类数
    if(cnt!=k)return 0;
    for(int i=1;i<=v;i++){//遍历每个顶点
        for(int j=h[i];j!=-1;j=ne[j]){//遍历每个顶点所连接的顶点
            int t=e[j];//t为顶点的序号
            if(color[i]==color[t])return 0;//如果相邻两点颜色相同，则不是一个解
        }
    }
    return 1;//所有点的颜色都符合要求，返回1
}

每一种方案，我们可以用stl库中的集合set存储，因为set自带去重功能，se.size()可以记录一共用了几种颜色

set包含在头文件 #include<set> 中

for(int i=0;i<n;i++){//n次查询
    set<int> se;//定义一个集合判断使用的颜色的种数
    for(int j=1;j<=v;j++){
        cin>>color[j];
        se.insert(color[j]);
    }
    if(judge(se.size()))cout<<"Yes"<<endl;
    else cout<<"No"<<endl;
}

C++代码如下：

#include<iostream>
#include<cstring>
#include<set>
using namespace std;
const int N=510,M=N*N;//没告诉边数的范围,最大为N*N,对应两两之间都有一条边
int h[N],e[M],ne[M],idx;
int color[N];//记录每个顶点的颜色
int v,m,k,n;
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int judge(int cnt){//判断一个方案是否合理
    if(cnt!=k)return 0;
    for(int i=1;i<=v;i++){//遍历每个顶点
        for(int j=h[i];j!=-1;j=ne[j]){//遍历每个顶点所连接的顶点
            int t=e[j];//t为顶点的序号
            if(color[i]==color[t])return 0;//如果相邻两点颜色相同，则不是一个解
        }
    }
    return 1;//所有点的颜色都符合要求，返回1
}
int main(){
    cin>>v>>m>>k;
    memset(h,-1,sizeof(h));//初始化h数组
    while(m--){//建图
        int a,b;
        cin>>a>>b;
        add(a,b),add(b,a);//无向图
    }
    cin>>n;
    for(int i=0;i<n;i++){//n次查询
        set<int> se;//定义一个集合判断使用的颜色的种数
        for(int j=1;j<=v;j++){
            cin>>color[j];
            se.insert(color[j]);
        }
        if(judge(se.size()))cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}