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;
}
版权声明:本文为jpwang8原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。