Untrusted Patrol

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
#include <queue>
using namespace std;
int order[100010],G[100010];
map<int,vector<int> > pt;
bool bfs(int x)
{
	bool mark=false;
	queue<int> t;
	t.push(x);
	while(!t.empty()){
		int pos=t.front();
		for(int i=0;i<pt[pos].size();i++){
			if(G[pt[pos][i]]>G[pos]) continue;
			else if(G[pt[pos][i]]==G[pos]) continue;
			else if(G[pt[pos][i]]==0) {G[pt[pos][i]]=G[pos];t.push(pt[pos][i]);}
			else mark=true;
		}
		t.pop();
	}
	return mark;
}
void bfs1(int x)
{
	queue<int> t;
	t.push(x);
	while(!t.empty()){
		int pos=t.front();
		for(int i=0;i<pt[pos].size();i++){
			if(G[pt[pos][i]]>G[pos]) continue;
			else if(G[pt[pos][i]]==G[pos]) continue;
			else {G[pt[pos][i]]=G[pos];t.push(pt[pos][i]);}
		}
		t.pop();
	}
}
int main()
{
	freopen("in.txt","r",stdin);
	int ca;
	cin>>ca;
	while(ca--){
		memset(G,0,sizeof(G));
		pt.clear();
		int n,m,k,t;
		cin>>n>>m>>k;
		for(int i=1;i<=k;i++) cin>>t;
		for(int i=1;i<=m;i++){
			int a,b;
			cin>>a>>b;
			if(!pt.count(a)) pt[a]=vector<int>();
			pt[a].push_back(b);
			if(!pt.count(b)) pt[b]=vector<int>();
			pt[b].push_back(a);
		}
		int l;
		cin>>l;
		for(int i=1;i<=l;i++){
			cin>>t;
			order[i]=t;
			G[t]=i;
		}
		if(l<k) {cout<<"No"<<endl;continue;}
		bfs1(order[1]);
		bool mark=true;
		for(int i=2;i<=l;i++){
			if(!bfs(order[i])) {mark=false;break;}
		}
		for(int i=1;i<=n;i++) if(!G[i]){
			mark=false;break;
		}
		if(mark) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}


还有一个不知道为什么不对的代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <vector>
using namespace std;
bool sensor[100010];
int order[100010],set[100010];
map<int,vector<int> > pt;
int gf(int x)
{
	if(set[x]==x) return x;
	set[x]=gf(set[x]);
	return set[x];
}
int main()
{
	int ca;
	cin>>ca;
	while(ca--){
		memset(sensor,false,sizeof(sensor));
		pt.clear();
		int n,m,k,t,l;
		scanf("%d%d%d",&n,&m,&k);
		for(int i=1;i<=k;i++){
			scanf("%d",&t);
			sensor[t]=true;
		}
		for(int i=1;i<=n;i++) set[i]=i;
		for(int i=1;i<=m;i++){
			int a,b;
			scanf("%d%d",&a,&b);
			int fa=gf(a),fb=gf(b);
			if(!sensor[fa]&&!sensor[fb])
				set[fa]=fb;
			else if(sensor[fa]&&!sensor[fb])
				set[fb]=fa;
			else if(!sensor[fa]&&sensor[fb])
				set[fa]=fb;
			else{
				if(!pt.count(fa)) pt[fa]=vector<int>();
				pt[fa].push_back(fb);
				if(!pt.count(fb)) pt[fb]=vector<int>();
				pt[fb].push_back(fa);
			}
		}
		scanf("%d",&l);
		for(int i=1;i<=l;i++){
			scanf("%d",&t);
			order[i]=t;
		}
		if(l!=k) {cout<<"No"<<endl;continue;}
		int root=order[1];
		bool mark=true;
		sensor[order[1]]=false;
		for(int i=2;i<=l;i++){
			int len=pt[order[i]].size();
			bool is=false;
			for(int j=0;j<len;j++){
				if(sensor[pt[order[i]][j]]) continue;
				t=gf(pt[order[i]][j]);
				if(t==root){
					set[order[i]]=root;
					sensor[order[i]]=false;
					is=true;
					break;
				}
			}
			if(!is) {mark=false;break;}
		}
		int judge=0;
		for(int i=1;i<=n;i++) if(set[i]==i)
			judge++;
		if(judge!=1) mark=false;
		if(mark) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值