10.02 T2 树上DP

Description

2.1 题目背景
银企鹅非常擅长化学。有一天他在试图命名一个巨大的单烯烃分子的时候，想到了一个问题。
2.2 题目描述
给你一棵树，一些边有标记，对于每条有标记的边，在树中找到包含这条边的一条最长链，并输出长度。

Input

第一行一个整数 id 表示测试点的编号。
多组数据，第二行一个整数 T 表示数据组数。
对于每组数据，第一行两个整数 n,m 表示节点的个数，和被标记的边的个数。
我们规定 1 是根，第二行 n − 1 个整数给出 2 ∼ n 父亲的编号，保证f(ai) < i。
第三行 m 个整数范围在 [2,n] 表示哪个点的父边被标记过。

Output

对于每组数据输出一行 m 个整数，必须与输入的边顺序一致，给出的是在这条边必选的情况下树中最长链的长度。

Sample Input

0 1 10 3 1 2 3 1 4 6 7 3 8 10 7 9

Sample Output

8 8 6

Hint

Source

root

 

 

 

一道很裸的树形DP，可惜我求了直径就直接爆炸了

code：

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 1000005
using namespace std;
struct node{
    int u,v;
}e[N];
int first[N],nxt[N],cnt,u[N],d1[N],d2[N],fa[N],c1[N],c2[N];
void add(int u,int v){
    e[++cnt].u=u;
    e[cnt].v=v;
    nxt[cnt]=first[u];
    first[u]=cnt;
}
void dfs1(int x){
    for(int i=first[x];i;i=nxt[i]){
        int v=e[i].v;
        if(v==fa[x])continue;
        dfs1(v);
        if(d1[v]+1>d1[x]){
            d2[x]=d1[x];
            d1[x]=d1[v]+1;
            c1[x]=v;
        }
        else {
            d2[x]=max(d2[x],d1[v]+1);
        }
    }
}
void dfs2(int x){
    if(fa[x]){//父亲不是1 
        if(c1[fa[x]]==x){
            u[x]=max(u[fa[x]],d2[fa[x]])+1;// 父亲向上的最大值和父亲的次大值 
        }
        else u[x]=max(u[fa[x]],d1[fa[x]])+1;//否则肯定就是父亲向下的最大值，当然还要加上1 
    }
    for(int i=first[x];i;i=nxt[i]){
        int v=e[i].v;
        if(v==fa[x])continue;
        dfs2(v);
    }
}
void init(){
    memset(e,0,sizeof e);
    memset(first,0,sizeof first);
    memset(nxt,0,sizeof nxt);
    cnt=0;
    memset(u,0,sizeof u);
    memset(d1,0,sizeof d1);
    memset(d2,0,sizeof d2);
    memset(c1,0,sizeof c1);
}
inline int Get(){
    register int rec=0;
    register char c;
    register bool f=0;
    for(c=getchar();c<'0'||c>'9';c=getchar())if(c=='-')f=true;
    while(c>='0'&&c<='9'){
        rec=(rec<<3)+(rec<<1)+c-'0';
        c=getchar();
    }
    return f?(-rec):rec;
}
int main(){
//    freopen("olefin.in","r",stdin);
    //freopen("olefin.out","w",stdout);
    int id;
    cin>>id;
    int T;
    T=Get();
    while(T--){
        init();
        int n,m;
        n=Get();m=Get();
        for(int i=2;i<=n;i++){
            fa[i]=Get();
            add(i,fa[i]);
            add(fa[i],i);
        }
        dfs1(1);
        dfs2(1);
        for(int i=1;i<=m;i++){
            int x;x=Get();
            cout<<d1[x]+u[x]<<" ";
        }
        if(m)cout<<'\n';
    }
    return 0;
}

over