hdu5735

很美妙的一题

官方题解 http://www.cnblogs.com/duoxiao/p/5777632.html

感觉有meet in middle的思想

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
const int mo=1e9+7;
struct way{int po,next;} e[400010];
unsigned int mx[260][260],f[70010],tmp[70010][260];
int n,w[70010],fa[70010],p[70010],v[260],ans,len;
char s[10];
void add(int x,int y)
{
    e[++len].po=y;
    e[len].next=p[x];
    p[x]=len;
}

void inc(int &a,int b)
{
    a+=b;
    if (a>mo) a-=mo;
}

int calc(int a,int b)
{
    if (s[0]=='X') return a^b;
    else return a&b;
}

void get(int x)
{
    if (x==1) f[x]=0;
    else f[x]=f[fa[x]]+(w[x]|w[fa[x]]);
    inc(ans,(ll)x*(f[x]+w[x])%mo);
    for (int i=p[x]; i; i=e[i].next) get(e[i].po);
}

void dfs(int x)
{
    f[x]=0;
    int a=w[x]>>8,b=w[x]&255;
    for (int i=0; i<256; i++)
        if (v[i]) f[x]=max(f[x],mx[i][b]+(calc(a,i)<<8));
    inc(ans,(ll)x*(f[x]+w[x])%mo);
    v[a]++;
    for (int i=0; i<256; i++)
    {
        tmp[x][i]=mx[a][i];
        mx[a][i]=max(mx[a][i],f[x]+calc(b,i));
    }
    for (int i=p[x]; i; i=e[i].next) dfs(e[i].po);
    memcpy(mx[a],tmp[x],sizeof(tmp[x]));
    v[a]--;
}

int main()
{
    int cas;
    scanf("%d",&cas);
    memset(mx,0,sizeof(mx));
    while (cas--)
    {
        len=0; memset(p,0,sizeof(p));
        scanf("%d%s",&n,s);
        for (int i=1; i<=n; i++) scanf("%d",&w[i]);
        for (int i=2; i<=n; i++)
        {
            scanf("%d",&fa[i]);
            add(fa[i],i);
        }
        ans=0;
        memset(v,0,sizeof(v));
        if (s[0]=='O') get(1);
        else dfs(1);
        for (int i=1; i<=n; i++) p[i]=0;
        printf("%d
",ans);
    }
}