[启发式合并 DP] Atcoder ARC086E. Smuggling Marbles

比赛的时候想到了每层独立，但是不会搞…

官方题解

好像学到了关于深度的信息启发式合并的次数是 O(n) 的…
还有deque这种神奇的东西……像我以前都是set加上各种外层的标记……

果然还是太弱了

#include <cstdio>#include <iostream>#include <algorithm>#include <queue>using namespace std;typedef pair<int,int> par;const int N=200010,P=1e9+7,inv2=P+1>>1;int n,cnt,G[N],dpt[N];struct edge{  int t,nx;}E[N];struct tp{  int x,y,z;  tp(){}  tp(int a,int b,int c):x(a),y(b),z(c){}};inline void addedge(int x,int y){  E[++cnt].t=y; E[cnt].nx=G[x]; G[x]=cnt;}inline int Pow(int x,int y){  int ret=1;  for(;y;y>>=1,x=1LL*x*x%P) if(y&1) ret=1LL*ret*x%P;  return ret;}deque<tp> f[N];void dfs(int x){  int mx=0,smx=0;  for(int i=G[x];i;i=E[i].nx){    dfs(E[i].t); int v=E[i].t;    if(f[v].size()>mx) smx=mx,mx=f[v].size();    else smx=max(smx,(int)f[v].size());    if(f[v].size()>f[x].size()) swap(f[x],f[v]);    for(int j=0;j<f[v].size();j++){      tp cur;      cur.x=(1LL*f[x][j].x*f[v][j].y+1LL*f[v][j].x*f[x][j].y)%P;      cur.y=1LL*f[x][j].y*f[v][j].y%P;      cur.z=(1LL*f[x][j].x*f[v][j].x+1LL*f[x][j].x*f[v][j].z+1LL*f[x][j].y*f[v][j].z+1LL*f[x][j].z*f[v][j].x+1LL*f[x][j].z*f[v][j].y+1LL*f[x][j].z*f[v][j].z)%P;      f[x][j]=cur;    }  }  for(int i=0;i<smx;i++){    f[x][i].y=(f[x][i].y+f[x][i].z)%P;    f[x][i].z=0;  }  f[x].push_front(tp(inv2,inv2,0));}int main(){  scanf("%d",&n);  for(int i=1,x;i<=n;i++)    scanf("%d",&x),addedge(x,i);  dfs(0); int ans=0;  for(int i=0;i<f[0].size();i++)    ans=(ans+f[0][i].x)%P;  ans=1LL*ans*Pow(2,n+1)%P;  printf("%d\n",ans);  return 0;}