HYSBZ-5485【思维】[Usaco2018 Dec]The Cow Gathering

HYSBZ-5485[Usaco2018 Dec]The Cow Gathering

题目

◇题目传送门◆

题目大意

给定一棵树与一些限制$(u,v)$，要求从叶节点开始删除节点，并且节点$u$要比节点$v$先删除，求哪些点可能成为最后删的点。

思路

我们可以按如下方法找一个可行的点：

每次寻找一个没有限制的叶节点删去，然后不断重复上述操作直至只剩一个点，就可以得到一个可行点。

再分析一下就可以发现，如果有操作不成功的情况，则该数据无解。

多画几个图就可以发现，限制$(u,v)$若存在，则以$v$为根的树中，以$u$为根的子树（包括$u$）均不能成为可行的点。

我们假设已经找出了一个可行的点$r$。

则我们可以把$r$提到根，此时我们可以发现，对于任意一个限制$(u,v)$，以$u$为根的子树（包括$u$）都是不合法的。

然后就可以发现，标记完所有点之后，合法的点都是连通的。

所以找到合法的点只需一次DFS即可。

所以，我们考虑对一个限制$(u,v)$，增加一条有向边$u$->$v$来表示这个限制。

则我们可以先将入度为$1$的点找出来，并以他们为起点做一次BFS，每次找到一个新的点就把这个点的入度减一，直到找到第一个入度为0的点，这时即可返回这个点为根。

参考代码

#include<queue>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;

const int Maxn=1e5;

vector<int> G[Maxn+5],G1[Maxn+5];
inline void addedge(int u,int v) {
	G[u].push_back(v);
	G[v].push_back(u);
}
int N,M;
int d[Maxn+5];
bool vis[Maxn+5];
int ans[Maxn+5];

int FindRoot() {
	int cnt=0;
	queue<int> q;
	for(int i=1;i<=N;i++)
		if(d[i]==1)q.push(i);
	while(!q.empty()) {
		int u=q.front();q.pop();
		++cnt;
		if(d[u]!=1) {
			if(d[u]<1)return u;
			else return -1;
		}
		for(int i=0;i<G[u].size();i++) {
			int v=G[u][i];
			d[v]--,cnt++;
			if(d[v]==1)q.push(v);
		}
		for(int i=0;i<G1[u].size();i++) {
			int v=G1[u][i];
			d[v]--,cnt++;
			if(d[v]==1)q.push(v);
		}
	}
	if(cnt!=N)return -1;
}

void DFS(int u,int fa) {
	ans[u]=1;
	for(int i=0;i<G[u].size();i++) {
		int v=G[u][i];
		if(vis[v]||v==fa)
			continue;
		DFS(v,u);
	}
}

int main() {
	#ifdef LOACL
	freopen("in.txt","r",stdin);
	freopen("out.txt","w",stdout);
	#endif
	scanf("%d %d",&N,&M);
	for(int i=1;i<N;i++) {
		int u,v;
		scanf("%d %d",&u,&v);
		addedge(u,v);
		d[u]++,d[v]++;
	}
	for(int i=1;i<=M;i++) {
		int u,v;
		scanf("%d %d",&u,&v);
		G1[u].push_back(v);
		d[v]++,vis[u]=true;
	}
	int rt=FindRoot();
	if(rt==-1) {
		for(int i=1;i<=N;i++)
			puts("0");
		return 0;
	}
	DFS(rt,-1);
	for(int i=1;i<=N;i++)
		printf("%d\n",ans[i]);
	return 0;
}