2020年7月 pat 甲級 第三題

題意：給出各個動物園之間的相連情況，要求每兩個相連動物園的動物種類不同 。總共有K種動物，n個動物園，r條路徑資訊。

#include<iostream>
#include<set>
#include<vector>
using namespace std;
struct Node{
	int animal;
	vector<int> nei;
}node[10000];
int n,r,k;
int main(){
	cin>>n>>r>>k;
	getchar();
	int a,b;
	for(int i=0;i<r;i++){
		scanf("%d %d",&a,&b);
		node[a].nei.push_back(b);
		node[b].nei.push_back(a);
	}
	int m,ani;
	cin>>m;
	for(int i=0;i<m;i++){
        set<int> se;
		bool flag=true;
		for(int j=1;j<=n;j++){
			scanf("%d",&ani);
			node[j].animal=ani;
			se.insert(ani);
		}
		if(se.size()>k) printf("Error: Too many species.\n");
		else if(se.size()<k) printf("Error: Too few species.\n");
		else{
			for(int j=1;j<=n;j++){
				for(int t=0;t<node[j].nei.size();t++){
                    int temp=node[j].nei[t];
					if(node[j].animal==node[temp].animal)
					{
						flag=false;
						break;
					}
				}
				if(flag==false) break;
			}
			if(flag==false) printf("No\n");
			else printf("Yes\n");
		}
	}
    return 0;
}