JZOJ5448. 【NOIP2017提高A组冲刺11.3】机房比教室好多了

Description

这里有一个N 个点的树, 节点从1 到N 编号, 第i 条边连接了ai 和bi.
一开始第i 个点上有Ai 个石头. Takahashi 和Aoki 会玩一个游戏.
首先, Takahashi 会选择一个出发点; 然后, 从Takahashi 开始, 他们会轮流进行如下操作:
• 首先, 从当前的点上拿走一个石头.
• 然后, 走到一个相邻的点上.
当一个玩家无法执行操作时, 他就输了. 请你找到所有可以让Takahashi 获胜的出发点.

Input

N
A1 A2 ::: AN
a1 b1 …
aN−1 bN−1

Output

按升序输出一行表示可以让Takahashi 获胜的出发点.

Sample Input

输入1：
3
1 2 3
1 2
2 3
输入2：
5
5 4 1 2 3
1 2
1 3
2 4
2 5
输入3：
3
1 1 1
1 2
2 3

Sample Output

输出1：
2
输出2：
1 2
输出3：
(输出为空)

Data Constraint

对于30% 的数据, N <=7;Ai <= 6;
对于60% 的数据, N <= 2000;
对于100% 的数据, 1 <= N <= 10^6, 1 <= ai, bi <= N; 0 <= Ai <= 10^9, 保证输入的是一棵树.

题解

这题是博弈，对于每一个节点，设一个状态，$f_i$表示它先手的胜败状态。
考虑最优策略：
假设当前在x号点，如果它有一个儿子y，满足$a_y<a_x$，那么后手就不可能重新再去取x父亲或者更上面的点的a，因为每一次从x回到y的时候，先手都可以从x走回到y。
如果这个儿子是y是先手必败的，你们x就是先手必胜的。

这样就有了一个n方dp了。
但是这样还是不行，其实在更换树的根的时候，很多操作都是重复的，
对于一个根节点root，它的一个儿子u，如果将u变为根节点，
仅仅是改变了一条边的情况，随之修改的也只有root的胜败状态和u的胜败状态。
这样的换根操作可以优化到O(n)

code

#include <iostream>
#include <cstdio>
#include <cmath>
#include <math.h>
#include <cstring>
#include <algorithm>
#define N 1000003
using namespace std;
char ch;
void read(int& n)
{
    n=0;
    ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while('0'<=ch && ch<='9')n=(n<<1)+(n<<3)+ch-'0',ch=getchar();
}
void write(int x)
{
    if(x>9)write(x/10);
    putchar(x%10+48);

}
int max(int x,int y){return x>y?x:y;}


int n,a[N],x,y,f[N],ans[N];
int nxt[N*2],to[N*2],b[N],tot;

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

void exchange(int x/*root*/,int y/*new root*/)
{
    if(a[x]>a[y] && f[y]==0)f[x]--;
    if(f[x]==0 && a[x]<a[y])f[y]++;
}

void dfs(int x)
{
    f[x]=0;
    for(int i=b[x];i;i=nxt[i])
        if(f[to[i]]<0)
        {
            dfs(to[i]);
            if(a[x]>a[to[i]] && f[to[i]]==0)f[x]++;
        }
}

void work(int x,int fa)
{
    if(f[x]>0)ans[++ans[0]]=x;
    for(int i=b[x];i;i=nxt[i])
        if(to[i]!=fa)
        {
            exchange(x,to[i]);
            work(to[i],x);
            exchange(to[i],x);
        }
}

int main()
{
    freopen("c.in","r",stdin);
    freopen("c.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),ins(y,x);

    memset(f,128,sizeof(f));
    dfs(1);
    work(1,0);
    sort(ans+1,ans+1+ans[0]);
    for(int i=1;i<=ans[0];i++)
        write(ans[i]),putchar(' ');

    return 0; 
}