topcoder13185 TreePuzzle

https://community.topcoder.com/stat?c=problem_statement&pm=13185

被wck屠了。

考试时候想分类讨论，结果发现情况有点复杂，最后还是没调出来。

回去看了看题解，发现好像是树形DP，状态记得很巧妙。

假设当前红点在$x$，从$fa$来，容易知道此时$fa$是空的。

容易知道以$fa$为根的子树（即如图的绿色圈）中的黑点是可以任意移动的。

因为$fa$是空的，所以我们可以先把红点从$x$移到$fa$，然后将以$x$为子树中的所以黑点任意移动，最后再把红点从$fa$移到$x$。

所以以$x$为根的子树（即如图的蓝色圈）中的黑点也是可以任意移动的。

所以我们只需要知道以$x$为根的子树中的黑点的个数即可。

记$vis[x][fa][c]$表示当前红点在$x$，从$fa$来，且以$x$为根的子树中的黑点的个数为$c$的状态，$1$表示可以到达这种状态，$0$表示不可以到达这种状态。

然后DP即可。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<utility>
#include<set>
#include<bitset>
#include<vector>
#include<functional>
#include<deque>
#include<cctype>
#include<climits>
#include<complex>
//#include<bits/stdc++.h>适用于CF,UOJ，但不适用于poj

using namespace std;

typedef long long LL;
typedef double DB;
typedef pair<int,int> PII;
typedef complex<DB> CP;

#define mmst(a,v) memset(a,v,sizeof(a))
#define mmcy(a,b) memcpy(a,b,sizeof(a))
#define fill(a,l,r,v) fill(a+l,a+r+1,v)
#define re(i,a,b)  for(i=(a);i<=(b);i++)
#define red(i,a,b) for(i=(a);i>=(b);i--)
#define ire(i,x) for(typedef(x.begin()) i=x.begin();i!=x.end();i++)
#define fi first
#define se second
#define m_p(a,b) make_pair(a,b)
#define p_b(a) push_back(a)
#define SF scanf
#define PF printf
#define two(k) (1<<(k))

template<class T>inline T sqr(T x){return x*x;}
template<class T>inline void upmin(T &t,T tmp){if(t>tmp)t=tmp;}
template<class T>inline void upmax(T &t,T tmp){if(t<tmp)t=tmp;}

inline int sgn(DB x){if(abs(x)<1e-)return ;return(x>)?:-;}
const DB Pi=acos(-1.0);

int gint()
  {
        int res=;bool neg=;char z;
        for(z=getchar();z!=EOF && z!='-' && !isdigit(z);z=getchar());
        if(z==EOF)return ;
        if(z=='-'){neg=;z=getchar();}
        for(;z!=EOF && isdigit(z);res=res*+z-'',z=getchar());
        return (neg)?-res:res;
    }
LL gll()
  {
      LL res=;bool neg=;char z;
        for(z=getchar();z!=EOF && z!='-' && !isdigit(z);z=getchar());
        if(z==EOF)return ;
        if(z=='-'){neg=;z=getchar();}
        for(;z!=EOF && isdigit(z);res=res*+z-'',z=getchar());
        return (neg)?-res:res;
  }

const int maxn=;

int n;
int d[maxn+],g[maxn+][maxn+];
int mark[maxn+];
int sz[maxn+][maxn+];//sz[i][j]表示i的父亲为j时，子树i的大小
int bl[maxn+];
int res[maxn+];

void addedge(int u,int v){g[u][++d[u]]=v;}

int calsz(int x,int fa)
  {
      int &res=sz[x][fa],j;
      if(res)return res;
      res=;
      re(j,,d[x])if(g[x][j]!=fa)res+=calsz(g[x][j],x);
      return res;
  }
int calbl(int x,int fa)
  {
      int &res=bl[x],j;
      res=mark[x];
      re(j,,d[x])if(g[x][j]!=fa)res+=calbl(g[x][j],x);
      return res;
  }

queue<int>Q;
int vis[maxn+][maxn+][maxn+];
void push(int x,int fa,int c)
  {
      if(vis[x][fa][c])return;
      vis[x][fa][c]=;
      Q.push((x<<)|(fa<<)|c);
  }

int main()
  {
      freopen("puzzle.in","r",stdin);
      freopen("puzzle.out","w",stdout);
      int i,j;
      n=gint();
      re(i,,n){int fa=gint()+;if(fa)addedge(fa,i),addedge(i,fa);}
        re(i,,n)re(j,,d[i])calsz(i,g[i][j]);
      re(i,,n)mark[i]=gint();mark[]=;
      calbl(,-);
      int total=bl[];
      re(j,,d[])
        {
            int v=g[][j];
            if(sz[v][]>bl[v])push(v,,bl[v]);
        }
      res[]=;
      while(!Q.empty())
        {
            int status=Q.front(),x=status>>,fa=(status>>)&,c=status&;Q.pop();//当前红点在x，从fa来，子树内的黑点数为c
            res[x]=;
            if(total-c<sz[fa][x])push(fa,x,total-c);
            int sc=;
            re(j,,d[x])if(g[x][j]!=fa)sc+=sz[g[x][j]][x];
            re(j,,d[x])if(g[x][j]!=fa)
              {
                  int v=g[x][j];
                  for(int c2=;c2<=c && c2<sz[v][x];c2++)if(c-c2<=sc-sz[v][x])push(v,x,c2);
              }
        }
      re(i,,n)PF("%d ",res[i]);PF("\n");
      return ;
  }