A1149 Dangerous Goods Packaging (25)

写题的时候有点着急了,其实对于fill这个数组可以有更好的处理

#include
#include
#include
using namespace std;
const int maxn=1001000;
bool vis[maxn]={false};
vector<int>G[maxn];
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=0;i<n;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		G[a].push_back(b);
		G[b].push_back(a);
	}
	int k,a;
	for(int i=0;i<m;i++)
	{
		scanf("%d",&k);
		bool flag=true;
		fill(vis,vis+maxn,false);
		for(int j=0;j<k;j++)
		{
			scanf("%d",&a);
			if(vis[a]==true)
			{
				flag=false;
			}
			else
			{
				vis[a]=true;
			}
			for(int l=0;l<G[a].size();l++)
			{
				vis[G[a][l]]=true;
			}
		}
		if(flag)
		{
			printf("Yes\n");
		}
		else
		{
			printf("No\n");
		}
	}
}

你可能感兴趣的:(pat刷题)