jzoj3501 【NOIP2013模拟联考15】消息传递(news) 树形dp

Description

---

H国的社会等级森严，除了国王之外，每个人均有且只有一个直接上级，当然国王没有上级。如果A是B的上级，B是C的上级，那么A就是C的上级。绝对不会出现这样的关系：A是B的上级，B也是A的上级。

最开始的时刻是0，你要做的就是用1单位的时间把一个消息告诉某一个人，让他们自行散布消息。在任意一个时间单位中，任何一个已经接到消息的人，都可以把消息告诉他的一个直接上级或者直接下属。

现在，你想知道：

1.到底需要多长时间，消息才能传遍整个H国的所有人？
2.要使消息在传递过程中消耗的时间最短，可供选择的人有那些？

Data Constraint

---

对于20%的数据，满足N<=3000；
对于50%的数据，满足N<=20000；
对于100%的数据，满足N<=200000；

Solution

---

一个朴素的想法就是枚举根然后做dp，显然会t
换个思路，可以先定1为根算出答案，再通过已知的某些条件换根
设f[i]表示以i为根的子树的答案，g[i]表示除以i为根的子树外的答案。设当前点为x，父亲点为fa，则我们需要求出fa不要x这棵子树的答案。考虑到之前统计答案的时候按大小排序后分配了权重均可表示为f[j]+rank[j]，设x在序列中位置为pos，那么1~pos-1都是不受影响的，pos+1之后的均少1（去掉了一位），用前缀后缀最大值维护即可

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;i++)
#define drp(i,st,ed) for (int i=st;i>=ed;i--)
#define erg(i,st) for (int i=ls[st];i;i=e[i].next)
#define fill(x,t) memset(x,t,sizeof(x))
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
#define INF 0x7fffffff
#define N 401000
#define E N*2+1
struct edge{int x,y,next;}e[E];
int mx[2][N],v[N],f[N],g[N],tmp[N];
int ans[N],q[N],fa[N];
int ls[N],edCnt=0;
int n;
void read(int &x) {
    x=0; char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
}
void addEdge(int x,int y) {
    e[++edCnt]=(edge){x,y,ls[x]}; ls[x]=edCnt;
    e[++edCnt]=(edge){y,x,ls[y]}; ls[y]=edCnt;
}
bool cmp(int a,int b) {
    return v[a]<v[b];
}
void solve() {
    int h=1,t=0;
    q[++t]=1;
    while (h<=t) {
        int now=q[h++];
        erg(i,now) if (e[i].y!=fa[now]) {
            q[++t]=e[i].y;
        }
    }
    drp(i,n,1) {
        int cnt=0;
        int now=q[i];
        for (int j=ls[now];j;j=e[j].next) 
            if (e[j].y!=fa[now]) tmp[++cnt]=f[e[j].y];
        std:: sort(tmp+1,tmp+cnt+1);
        rep(j,1,cnt) f[now]=max(f[now],tmp[j]+cnt-j+1);
    }
    ans[1]=f[1];
    rep(i,1,n) {
        int cnt=0;
        int now=q[i];
        for (int j=ls[now];j;j=e[j].next)
            if (e[j].y!=fa[now]) tmp[++cnt]=e[j].y,v[e[j].y]=f[e[j].y];
        if (now!=1) tmp[++cnt]=fa[now],v[fa[now]]=g[now];
        std:: sort(tmp+1,tmp+cnt+1,cmp);
        mx[0][0]=mx[1][cnt+1]=0;
        rep(j,1,cnt) ans[now]=max(ans[now],v[tmp[j]]+cnt-j+1);
        rep(j,1,cnt) mx[0][j]=max(mx[0][j-1],v[tmp[j]]+cnt-j);
        drp(j,cnt,1) mx[1][j]=max(mx[1][j+1],v[tmp[j]]+cnt-j+1);
        rep(j,1,cnt) if (tmp[j]!=fa[now]) g[tmp[j]]=max(mx[0][j-1],mx[1][j+1]);
    }
}
int main(void) {
    freopen("news.in","r",stdin);
    freopen("news.out","w",stdout);
    read(n);
    rep(i,2,n) {
        read(fa[i]);
        addEdge(fa[i],i);
    }
    solve();
    int prt=INF;
    rep(i,1,n) prt=min(prt,ans[i]);
    printf("%d\n",prt+1);
    rep(i,1,n) if (ans[i]==prt) {
        printf("%d ", i);
    }
    return 0;
}