View Code of Problem 7

#include<cstdio>
#include<map>
#include<vector>
using namespace std;

int main(){
	//freopen("7.txt","r",stdin);
	int t,n,m;
	scanf("%d",&t);
	for(int i=0;i<t;i++){
		scanf("%d",&n);
		map<int,int> mp;
		for(int j=0;j<n;j++){
			int a,b;
			char c;
			scanf("%d %c %d",&a,&c,&b);
			mp[a]=b;mp[b]=a;			
		}
		scanf("%d",&m);
		vector<int> vt(m+1);
		for(int j=0;j<m;j++){
			scanf("%d",&vt[j]);
		}
		for(int j=0;j<m;j++){
			int flag=0;
			map<int,int>::iterator it=mp.begin();
			for(;it!=mp.end();it++){
				if(vt[j]==it->first||vt[j]==it->second){
					flag=1;break;
				}
			}
			if(flag==1)	printf("%d\n",mp[vt[j]]);
			else	printf("UNKNOW\n");
		}
				
	if(i!=t-1)	printf("\n");	
	}
	return 0;
} 

Double click to view unformatted code.


Back to problem 7