5438. 【NOIP2017提高A组集训10.31】Tree

T2要用到KMP，然后就屁颠屁颠地去问去查。
结果——被KMP弄到炸心态，还得多理解。
总结上次的T4，

---

我们可以考虑遍历树，但要知道的是，这题专卡$DFS$，险恶的出题人会设置爆栈RE。
所以我们看看$BFS$
我们可以设置一个数组为来表示一个节点被翻转的次数。如果它被翻转后还是黑的，那么考虑继续转。显然，满足题目条件的情况下，必定有解。
那么时间复杂度瞬间到了$O(n)$并且空间没有问题。
但我还是加上可快排，开始没有在建边的时候考虑去保证顺序这个问题，所以时间复杂度就上到了$O(nlogn)$（雾……）

#include<cstdio>
#include<cstring>
#include<iostream>
#define N 500005
using namespace std;
int n,a[N],ans[N],len=0;
int u,v,first[N],to[N],next[N],tot=0;
void add(int x,int y){to[++tot]=y,next[tot]=first[x],first[x]=tot;}
inline long long read()
{
	long long s=0,w=1;
	char str=getchar();
	while(str<'0'||str>'9')
	{
		if(str=='-') w=-1;
		str=getchar();
	}
	while(str>='0'&&str<='9')
		s=s*10+str-'0',str=getchar();
	return s*w;
}
void qsort(int l,int r)
{
	int i=l,j=r,mid=ans[(l+r)/2];
	while(i<=j){
		while(ans[i]<mid) i++;while(ans[j]>mid) j--;
		if(i<=j) swap(ans[i],ans[j]),i++,j--;	
	}if(i<r) qsort(i,r);if(l<j) qsort(l,j); 
}
void bfs()
{
	int head=0,tail=1,q[N][2];
	memset(q,0,sizeof(q)),q[1][0]=1;
	if(a[1]) q[1][1]=1,ans[++len]=1;
	while(head<tail)
	{
		head++;
		int from=q[head][0];
		for(int i=first[from];i;i=next[i])
		{
			q[++tail][0]=to[i];
			if((a[to[i]]+q[head][1])%2) q[tail][1]=q[head][1]+1,ans[++len]=to[i];
			else q[tail][1]=q[head][1];
		}
	}
}
int main()
{
	freopen("tree.in","r",stdin);
	freopen("tree.out","w",stdout);
	n=read();
	for(int i=1;i<=n;++i) a[i]=read();
	for(int i=1;i<n;++i) u=read(),v=read(),add(u,v);
	bfs(),qsort(1,len);
	for(int i=1;i<=len;++i) printf("%d ",ans[i]);
	return 0;
}