[USACO18DEC]The Cow Gathering

Description:

给定一棵树,每次删去叶子,有m个限制,分别为(a,b)表示a需要比b先删,为每个点能否成为最后被删的点

Hint:

\(n,m \le 10^5\)

Solution:

手模后会发现一个十分不显然的规律: 若a比b先删,则a在以b为根的子树中的点,都不能最后删,

于是这样就转化为这个题了:https://www.cnblogs.com/list1/p/10497877.html

每次直接打个差分标记

这题还要判无解的情况(这谁想得到啊)

#include <map>
#include <set>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define ls p<<1
#define rs p<<1|1
using namespace std;
typedef long long ll;
const int mxn=5e5+5;
int n,m,cnt,tot,in[mxn],hd[mxn],vis[mxn],hd2[mxn],sz[mxn],dep[mxn],ans[mxn],tag[mxn],dfn[mxn],f[mxn][19];
inline int read() {
	char c=getchar(); int x=0,f=1;
	while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
	while(c<='9'&&c>='0') {x=(x<<3)+(x<<1)+(c&15);c=getchar();}
	return x*f;
}
inline int chkmax(int &x,int y) {if(x<y) x=y;}
inline int chkmin(int &x,int y) {if(x>y) x=y;}

struct ed {
	int to,nxt;
}t[mxn<<1];

inline void add(int u,int v) {
	t[++cnt]=(ed) {v,hd[u]}; hd[u]=cnt;
}

inline void add2(int u,int v) {
    t[++cnt]=(ed) {v,hd2[u]}; hd2[u]=cnt;
    ++in[v];
}

int up(int u,int k)
{
	for(int i=30;i>=0;--i)
		if(k&(1<<i)) u=f[u][i];
	return u;
}

void dfs(int u,int fa)
{
	f[u][0]=fa; dfn[u]=++tot; dep[u]=dep[fa]+1; sz[u]=1;
	for(int i=hd[u];i;i=t[i].nxt) {
		int v=t[i].to;
		if(v==fa) continue ;
		dfs(v,u); sz[u]+=sz[v];
	}
}

void solve(int u,int fa,int val)
{
	val+=tag[u]; ans[u]=val>0;
	for(int i=hd[u];i;i=t[i].nxt) {
		int v=t[i].to;
		if(v==fa) continue ;
		solve(v,u,val);
	}
}

int flag=0;

void check()
{
	queue<int > q;
	for(int i=1;i<=n;++i)
		if(in[i]<2) q.push(i),vis[i]=1;
		while(!q.empty()) {
			int u=q.front(); q.pop();
			for(int i=hd[u];i;i=t[i].nxt) {
				int v=t[i].to; --in[v];
				if(!vis[v]&&in[v]<2) {
					vis[v]=1;
					q.push(v);
				}
			}
			for(int i=hd2[u];i;i=t[i].nxt) {
				int v=t[i].to; --in[v];
				if(!vis[v]&&in[v]<2) {
					vis[v]=1;
					q.push(v);
				}
			}
		}
	for(int i=1;i<=n;++i) if(!vis[i]) flag=1;
}

int main()
{
	n=read(); m=read(); int u,v,w;
	for(int i=1;i<n;++i) {
		u=read(); v=read();
		add(u,v); add(v,u);
		++in[u]; ++in[v];
	}
	dfs(1,1); f[1][0]=1;
	for(int j=1;j<=18;++j)
		for(int i=1;i<=n;++i)
			f[i][j]=f[f[i][j-1]][j-1];
	for(int i=1;i<=m;++i) {
		u=read(); v=read(); add2(u,v);
		if(dfn[v]>dfn[u]&&dfn[v]<dfn[u]+sz[u]) {
			int pos=up(v,dep[v]-dep[u]-1);
			--tag[pos];
			++tag[1];
		}
		else ++tag[u];
	}
	check();
	if(flag) {
		for(int i=1;i<=n;++i)
			printf("0\n");
		return 0;
	}
	solve(1,1,0);
	for(int i=1;i<=n;++i) printf("%d\n",ans[i]^1);
    return 0;
}