比赛 2026.3.28 评测结果 AAAAAAAAAWAAAAAAAAWAAAW
题目名称 Picking Flowers 最终得分 86
用户昵称 2_16鸡扒拌面 运行时间 17.068 s
代码语言 C++ 内存使用 64.00 MiB
提交时间 2026-03-28 11:06:37
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;

const int N=1000007,INF=0x3f3f3f3f;
int dp[N],dis[N],flw[N],dst[N];
bool vis[N],has[N];
vector<int> g[N],rev[N];

void bfs(){
	queue<int> q;
	q.push(1);
	dis[1]=0;
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=0;i<g[u].size();i++){
			int v=g[u][i];
			if(dis[v]==INF){
				dis[v]=dis[u]+1;
				q.push(v);
			}
		}
	}
}

int dfs(int u){
	if(dp[u]!=INF)return dp[u];
	dp[u]=0;
	for(int i=0;i<rev[u].size();i++){
		int v=rev[u][i];
		dp[u]|=dfs(v);
	}
	return dp[u];
}

int main(){
	freopen("Flowers.in","r",stdin);
	freopen("Flowers.out","w",stdout);
	int T;cin>>T;
	while(T--){
		int n,m,k,l,ok=0;
		cin>>n>>m>>k>>l;
		for(int i=1;i<=n;i++){
			g[i].clear();
			rev[i].clear();
			dp[i]=dis[i]=INF;
		}
		for(int i=1;i<=k;i++)cin>>flw[i];
		for(int i=1;i<=l;i++)cin>>dst[i];
		for(int i=1,u,v;i<=m;i++){
			cin>>u>>v;
			g[u].push_back(v);
			g[v].push_back(u);
		}
		for(int i=1;i<=n;i++)vis[i]=has[i]=0;
		bfs();
		for(int i=1;i<=n;i++){
			for(int j=0;j<g[i].size();j++){
				int v=g[i][j];
				if(dis[v]==dis[i]-1)rev[i].push_back(v);
			}
		}
		sort(flw+1,flw+k+1);
		sort(dst+1,dst+l+1);
		bool flag=1;
		for(int i=1;i<k;i++)if(dis[flw[i]]==dis[flw[i+1]])flag=0;
		if(flag){
			for(int i=1;i<=k;i++){
				vis[dis[flw[i]]]=1;
				has[flw[i]]=1;
			}
			for(int i=1;i<=n;i++)if(vis[dis[i]]&&!has[i])dp[i]=0;
			dp[1]=1;
			for(int i=1;i<=k;i++)if(!dfs(flw[i]))flag=0;
			if(flag){
				for(int i=1;i<=l;i++)if(dis[dst[i]]>=dis[flw[k]])ok|=dfs(dst[i]);
				if(!ok)flag=0;
			}
		}
		if(!flag){
			for(int i=2;i<=n;i++)dp[i]=0;
		}
		for(int i=2;i<=n;i++)if(dp[i]==INF)dp[i]=0;
		for(int i=2;i<=n;i++)cout<<dp[i];
		cout<<'\n';
	}
	return 0;
}