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

题目

题解

题目保证是存在解，
现在就考虑要得到一个合理的解，
每一个点有什么特别的特征。

如果一个点的颜色与其父亲的颜色不一样，
这个点就必须被标记。
如果这个点不被标记，无论它父亲怎样标记，都无法使得它们的颜色相同，

code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#define ll long long
#define N 1000003
#define db double
#define P putchar
#define G getchar
#define mo 1000000007
using namespace std;
char ch;
void read(int &n)
{
    n=0;
    ch=G();
    while((ch<'0' || ch>'9') && ch!='-')ch=G();
    ll w=1;
    if(ch=='-')w=-1,ch=G();
    while('0'<=ch && ch<='9')n=(n<<3)+(n<<1)+ch-'0',ch=G();
    n*=w;
}

int max(int a,int b){return a>b?a:b;}
ll min(ll a,ll b){return a<b?a:b;}
ll abs(ll x){return x<0?-x:x;}
ll sqr(ll x){return x*x;}
void write(ll x){if(x>9) write(x/10);P(x%10+'0');}
void writeln(ll x){write(x);P('\n');}
int gcd(int x,int y){return y==0?x:gcd(y,x%y);}

int a[N],nxt[N],b[N],to[N],tot,x,y,n,ans[N];
int d[N],c[N],l,r;

void ins(int x,int y)
{
    nxt[++tot]=b[x];
    to[tot]=y;
    b[x]=tot;
}

void dfs(int x,int color)
{
    if(a[x]!=color)ans[++ans[0]]=x;
    for(int i=b[x];i;i=nxt[i])
        dfs(to[i],a[x]);
}

int main()
{
    freopen("tree.in","r",stdin);
    freopen("tree.out","w",stdout);
    read(n);
    for(int i=1;i<=n;i++)
        read(a[i]);
    for(int i=1;i<n;i++)
        read(x),read(y),ins(x,y);
    //dfs(1,0); 
    d[r=1]=1;c[1]=l=0;
    while(l<r)
    {
        x=d[++l];
        if(a[x]!=c[l])ans[++ans[0]]=x;
        for(int i=b[x];i;i=nxt[i])
            d[++r]=to[i],c[r]=a[x];
    }
    sort(ans+1,ans+1+ans[0]);
    for(int i=1;i<=ans[0];i++)
        write(ans[i]),P(' ');
}