九校联考-DL24凉心模拟Day1T2 烯烃(olefin)

2.1 题目背景

银企鹅非常擅长化学。有一天他在试图命名一个巨大的单烯烃分子的时候，想到了一个问题。

2.2 题目描述

给你一棵树，一些边有标记，对于每条有标记的边，在树中找到包含这条边的一条最长链，并输出长度。

2.3 格式

2.3.1 输入格式

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

2.3.2 输出格式

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

2.4 样例

2.4.1 样例输入

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

2.4.2 样例输出

8 8 6

2.5 数据范围

测试点	n	m	T	特殊约定
1,2	100	\(<=n-1\)	100	无
3,4	105	10	100	无
5	105	\(<=n-1\)	100	树是一条链
6	105	\(<=n-1\)	100	所有fai = 1
7,8,9,10	\(10^5\)	\(<=n-1\)	100	无

树型dp，暴力做+特殊点（1,2,3,4+5,6）一共60分，dp出这个点下面/上面最大长度，次大长度，用根据标记边的位置用不同的值更新

#include <bits/stdc++.h>
#define enter putchar('\n')
#define space putchar(' ')
#define pii pair<int,int>
#define fi first
#define se second
#define MAXN 100005
#define pb push_back
#define ivorysi
using namespace std;
typedef long long int64;
typedef double db;
template<class T>
void read(T &res) {
    res = 0;T f = 1;char c = getchar();
    while(c < '0' || c > '9') {
        if(c == '-') f = -1;
        c = getchar();
    }
    while(c >= '0' && c <= '9') {
        res = res * 10 + c - '0';
        c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {x = -x;putchar('-');}
    if(x >= 10) out(x / 10);
    putchar('0' + x % 10);
}
int N,M,fa[MAXN];
int dp[MAXN],ans[MAXN],fr[MAXN];
struct node {
    int to,next;
}E[MAXN * 2];
int head[MAXN],sumE;
void add(int u,int v) {
    E[++sumE].to = v;
    E[sumE].next = head[u];
    head[u] = sumE;
}
void dfs1(int u) {
    for(int i = head[u] ; i ; i = E[i].next) {
        int v = E[i].to;
        dfs1(v);
        dp[u] = max(dp[v] + 1,dp[u]);
    }
}
void dfs2(int u) {
    pii p(0,0);
    for(int i = head[u] ; i ; i = E[i].next) {
        int v = E[i].to;
        ans[v] = max(dp[v] + 1 + fr[u],ans[v]);
        fr[v] = max(fr[v],fr[u] + 1);
        if(dp[v] >= dp[p.se]) p.se = v;
        if(dp[p.se] >= dp[p.fi]) swap(p.se,p.fi);
    }
    for(int i = head[u] ; i ; i = E[i].next) {
        int v = E[i].to;
        if(v != p.fi) {
            ans[v] = max(dp[v] + 2 + dp[p.fi],ans[v]);
            fr[v] = max(fr[v],dp[p.fi] + 2);
        }
        else if(p.se) {
            ans[p.fi] = max(ans[p.fi],dp[p.fi] + 2 + dp[p.se]);
            fr[p.fi] = max(fr[p.fi],dp[p.se] + 2);
        }
        dfs2(v);
    }
}
void Solve() {
    memset(head,0,sizeof(head));sumE = 0;
    memset(fr,0,sizeof(fr));memset(dp,0,sizeof(dp));memset(ans,0,sizeof(ans));
    read(N);read(M);
    for(int i = 2 ; i <= N ; ++i) {
        read(fa[i]);
        add(fa[i],i);
    }
    dfs1(1);dfs2(1);
    int x;
    for(int i = 1 ; i <= M ; ++i) {
        read(x);out(ans[x]);
        if(i == M) enter;
        else space;
    }
    if(!M) enter;
}
int main() {
    freopen("olefin.in","r",stdin);
    freopen("olefin.out","w",stdout);
    int id,T;
    read(id);read(T);
    while(T--) {
        Solve();
    }
    return 0;
}