题目描述
银企鹅非常擅长化学。有一天他在试图命名一个巨大的单烯烃分子的时候,想到了一个问题。
给你一棵树,一些边有标记,对于每条有标记的边,在树中找到包含这条边的一条最长链,并输出长度。
输入格式
第一行一个整数 id 表示测试点的编号。
多组数据,第二行一个整数 T 表示数据组数。
对于每组数据,第一行两个整数 n, m 表示节点的个数,和被标记的边的个数。
我们规定 1 是根,第二行 n-1 个整数给出 2到n 父亲的编号,保证 fai f a i < i。
第三行 m 个整数范围在 [2, n] 表示哪个点的父边被标记过。
输出格式
对于每组数据输出一行 m 个整数,必须与输入的边顺序一致,给出的是在这条边必选的情况下树中最长链的长度。
样例数据
input
0
1
10 3
1 2 3 1 4 6 7 3 8
10 7 9
output
8 8 6
数据规模与约定
时间限制: 3s 3 s
空间限制: 512MB 512 MB
(懒得分割了就这样吧
因为要过给定边
肯定是两个点的两边各找一条最长链
比如求过边
(x,y)
(
x
,
y
)
的最长链
令
fa[x]=y
f
a
[
x
]
=
y
x的一端很好处理
y的一端还需要往上处理一条最长
用往下的值再dfs一遍更新答案即可。
#include<bits/stdc++.h>
using namespace std;
#pragma GCC optimize(2)
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back
#define pc putchar
#define mp make_pai
#define file(k) memset(k,0,sizeof(k))
#define ll long long
namespace fastIO{
#define BUF_SIZE 100000
#define OUT_SIZE 100000
bool IOerror = 0;
inline char nc(){
static char buf[BUF_SIZE],*p1 = buf+BUF_SIZE, *pend = buf+BUF_SIZE;
if(p1 == pend){
p1 = buf; pend = buf+fread(buf, 1, BUF_SIZE, stdin);
if(pend == p1){ IOerror = 1; return -1;}
}
return *p1++;
}
inline bool blank(char ch){return ch==' '||ch=='\n'||ch=='\r'||ch=='\t';}
inline void read(int &x){
bool sign = 0; char ch = nc(); x = 0;
for(; blank(ch); ch = nc());
if(IOerror)return;
if(ch == '-') sign = 1, ch = nc();
for(; ch >= '0' && ch <= '9'; ch = nc()) x = x*10+ch-'0';
if(sign) x = -x;
}
inline void read(ll &x){
bool sign = 0; char ch = nc(); x = 0;
for(; blank(ch); ch = nc());
if(IOerror) return;
if(ch == '-') sign = 1, ch = nc();
for(; ch >= '0' && ch <= '9'; ch = nc()) x = x*10+ch-'0';
if(sign) x = -x;
}
#undef OUT_SIZE
#undef BUF_SIZE
};
using namespace fastIO;
int n , m;
int u , v;
int fa[101000],dep[101000];
bool spe[101000];
int mx[101000] , nx[101000] , fmx[101000];
int id_m[101000] , id_n[101000];
int linkk[101000] , t;
struct node{
int n , y;
}e[201000];
queue<int>q;
bool inq[101000];
int bfs(int S)
{
memset(inq,0,sizeof(inq));
inq[S] = true;q.push(S);
int id;
while(!q.empty())
{
int x = q.front();q.pop();
id = x;
rept(i,x)
if(!inq[e[i].y])
inq[e[i].y] = true,
q.push(e[i].y);
}
return id;
}
void dfss(int x)
{
dep[x] = dep[fa[x]] + 1;
mx[x] = nx[x] = id_n[x] = id_m[x] = 0;
int k = linkk[x];
if(x != 1 && e[k].n == 0)
{
id_m[x] = x;
return;
}
rept(i,x)
if(e[i].y != fa[x])
{
dfss(e[i].y);
if(mx[e[i].y]+1 > mx[x])
{
nx[x] = mx[x] , id_n[x] = id_m[x];
mx[x] = mx[e[i].y] + 1,id_m[x] = id_m[e[i].y];
}
else
if(mx[e[i].y]+1 > nx[x])
nx[x] = mx[e[i].y] + 1,id_n[x] = id_m[e[i].y];
}
}
void dfs1(int x,int Max)
{
// if(x % 1000 == 0)printf("#%d %d\n",x,Max);
if(x != 1) fmx[x] = Max;
rept(i,x)
{
int y = e[i].y;
if(y == fa[x]) continue;
bool flag;int det;
if(id_m[y] == id_m[x]) flag = false;
else flag = true;
if(Max != 0) Max > (flag?mx[x]:nx[x]) ? dfs1(y,Max+1) : dfs1(y,flag?mx[x]+1:nx[x]+1);
else dfs1(y,flag?mx[x]+1:nx[x]+1);
}
}
void insert(int x,int y)
{
e[++t].y = y;e[t].n = linkk[x];linkk[x] = t;
e[++t].y = x;e[t].n = linkk[y];linkk[y] = t;
}
void init()
{
read(n);read(m);t = 0;
rep(i,2,n)
{
read(fa[i]);
insert(i,fa[i]);
}
dfss(1);
dfs1(1,0);
rep(i,1,m)
{
int x;read(x);
int tmp = 0,Max = 0,y = fa[x];
if(id_m[y] != id_m[x]) Max = mx[y];
else Max = nx[y];
tmp = fmx[y];
Max = max(Max,tmp);
printf("%d ",Max+1+mx[x],i);
}
printf("\n");
rep(i,1,n) linkk[i] = 0;
}
int main()
{
freopen("olefin.in","r",stdin);
freopen("olefin.out","w",stdout);
int T;read(T);read(T);
while(T--)init();
return 0;
}