PAT甲级1154
题目大意:给出n个顶点,m条边。接下来给出k组数据,问每组数据是否能构成彩色图,要求任意两个相连的顶点颜色不能相同。如果不能输出No,否则输出一共多少种颜色。
这里后两个测试点超时,一开始以为是set问题,后来发现这题用邻接表数据依然会很大,改成储存边即可。尽量不要额外开辟一个visit判断多少种颜色,数据int范围会越界。
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
#define MAXN 10005
using namespace std;
struct edge{
int v1,v2;
};
int n,m;
vector<edge> e;
bool isColoring(vector<int> q){
for(int i=0;i<e.size();i++){
if(q[e[i].v1]==q[e[i].v2]) return false;
}
return true;
}
int main(){
int vertex1,vertex2;
scanf("%d%d",&n,&m);
for(int i=0;i<m;i++){
scanf("%d%d",&vertex1,&vertex2);
edge tmpedge;
tmpedge.v1=vertex1,tmpedge.v2=vertex2;
e.push_back(tmpedge);
}
int k,tmpv;
scanf("%d",&k);
vector<int> query;
set<int> color;
for(int i=0;i<k;i++){
color.clear();
query.clear();
for(int j=0;j<n;j++){
scanf("%d",&tmpv);
color.insert(tmpv);
query.push_back(tmpv);
}
if(isColoring(query)) printf("%d-coloring\n",color.size());
else printf("No\n");
}
system("pause");
return 0;
}