PAT甲级1154

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