[Codeforces 763A]Timofey and a tree

正解：dfs一遍然后根据边两边颜色不同乱搞。

可惜我不会。

想了想后发现，如果要求一个节点，使得所有子树纯色的话，把它搞成一个序列不就好办了。

于是就想到了欧拉序列——每一棵子树在欧拉序列上都是一段连续的空间。

那么就可以把这棵树映射到序列上了。

然后就可以暴力枚举每一个节点，然后暴力枚举所有子树了。

可以证明，这么做是O(n)的。

那么到查询了，也就是查询某段连续的区间是否完全相同（换句话说，都是一个数）。

据说可以线段树维护，可惜我写了好几遍都挂了。。

于是就搞了个分块，当然是自带剪枝的分块。

然后就水过去。。

（话说cf的数据好弱啊。。。）

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <vector>
#include <cmath>
using namespace std;

const int N=600000;
int h[N],to[N],r[N],c[N],tot;
void add(int u,int v){
    to[tot]=v;
    r[tot]=h[u];
    h[u]=tot++;
}

vector<int>dfn,color,G[N];

void dfs(int x,int f){
    G[x].push_back(dfn.size()+1);
    dfn.push_back(x);
    color.push_back(c[x]);
    for(int i=h[x];i!=-1;i=r[i]){
        if(to[i]!=f){
            dfs(to[i],x);
            G[x].push_back(dfn.size()+1);
            dfn.push_back(x);
            color.push_back(c[x]);
        }
    }
}

int n;

int sz,top,bl;
int st[N],id[N];
vector<int>bot[N];
bool pure[N];
void build(){
    for(int i=0;i<color.size();i++)st[++top]=color[i];
    sz=bl=sqrt(top);
    for(int i=1;i<=top;i++){id[i]=bot[(i-1)/sz+1].size();bot[(i-1)/sz+1].push_back(st[i]);}
    sz=(top-1)/bl+1;
    for(int i=1;i<=sz;i++){
        pure[i]=true;
        for(int j=0;j<bot[i].size();j++)
            if(j!=0&&bot[i][j]!=bot[i][j-1]){pure[i]=false;break;}
    }
    // for(int i=1;i<=sz;i++){
    //     for(int j=0;j<bot[i].size();j++){
    //         cout<<bot[i][j]<<' ';
    //     }
    //     cout<<endl;
    // }
}

bool isPure(int l,int r){
    if(l>r)return true;
    int lef=(l-1)/bl+1;
    int rig=(r-1)/bl+1;
    if(lef==rig){
        for(int i=l;i<=r;i++)
            if(i!=l&&st[i]!=st[i-1])return false;
    }else{
        vector<int>cls;
        for(int i=id[l];i<bot[lef].size();i++)cls.push_back(bot[lef][i]);
        for(int i=lef+1;i<=rig-1;i++){cls.push_back(bot[i][0]);if(!pure[i])return false;}
        for(int i=0;i<=id[r];i++)cls.push_back(bot[rig][i]);
        for(int i=0;i<cls.size();i++)if(i!=0&&cls[i]!=cls[i-1])return false;
        // for(int i=0;i<cls.size();i++)cout<<cls[i]<<' ';
        // cout<<endl;
        // printf("IDL:%d IDR:%d lef:%d rig:%d L:%d R:%d\n",id[l],id[r],lef,rig,l,r);
    }
    return true;
}

int main(){
    memset(h,-1,sizeof(h));
    scanf("%d",&n);
    for(int i=1,u,v;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    for(int i=1;i<=n;i++)scanf("%d",&c[i]);
    dfs(1,0);
    build();
    // for(int i=0;i<color.size();i++)cout<<color[i]<<' ';
    // cout<<endl;
    // cout<<isPure(3,5);
    // return 0;
    for(int i=1;i<=n;i++){
        for(int j=0;j<G[i].size()-1;j++)
            if(!isPure(G[i][j]+1,G[i][j+1]-1))goto nxt;
        if(isPure(1,G[i][0]-1)&&isPure(G[i][G[i].size()-1]+1,dfn.size())&&color[0]==color[color.size()-1])
            return 0&printf("YES\n%d\n",i);
        nxt:;    
    }
    puts("NO");
}