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

题目描述

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

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

输入输出格式

输入格式

第一行一个整数 $id$ 表示测试点的编号。

多组数据，第二行一个整数 $T$ 表示数据组数。

对于每组数据，第一行两个整数 $n$, $m$ 表示节点的个数，和被标记的边的个数。

我们规定 $1$ 是根，第二行 $n-1$ 个整数给出 $2\sim n$ 父亲的编号，保证$f{a}_i<i$。

第三行 $m$ 个整数范围在 $[2,n]$ 表示哪个点的父边被标记过。

输出格式

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

输入输出样例

输入样例#1：

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

输出样例#1：

8 8 6

数据范围

测试点	   n ≤	  m ≤	T ≤	     特殊约定
1, 2	 100	n − 1	100	     无
3, 4	 10^5	10	    100	     无
5	     10^5	n − 1	100	   树是一条链
6	     10^5	n − 1	100	   所有 fa_i=1
7, 8, 9,  10	10^5	n − 1	100	无

解题分析

直接$DFS$预处理出子树中最深和次深的深度， 再一遍$DFS$即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cctype>
#include <ctime>
#include <cmath>
#define R register
#define IN inline
#define File freopen("olefin.in", "r", stdin), freopen("olefin.out", "w", stdout)
#define W while
#define MX 100050
inline char nc()
{
	static const int buflen=1e6;
	static char buf[buflen],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,buflen,stdin),p1==p2)?EOF:*p1++;
}
#define gc nc()
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
int head[MX], ans[MX], dep[MX], fir[MX], sec[MX], sta[MX];
bool mark[MX];
struct Edge {int to, nex;} edge[MX << 1];
int dot, line, cnt;
IN void add(R int from, R int to) {edge[++cnt] = {to, head[from]}, head[from] = cnt;}
void DFS(R int now)
{
	if(!head[now]) return fir[now] = dep[now], void();
	for (R int i = head[now]; i; i = edge[i].nex)
	{
		dep[edge[i].to] = dep[now] + 1;
		DFS(edge[i].to);
		if(fir[now] < fir[edge[i].to]) sec[now] = fir[now], fir[now] = fir[edge[i].to];
		else if(sec[now] < fir[edge[i].to]) sec[now] = fir[edge[i].to];
	}
}
void calc(R int now, R int mx)
{
	for (R int i = head[now]; i; i = edge[i].nex)
	{
		R int del = mx;
		if(fir[edge[i].to] == fir[now]) del = std::max(sec[now] - dep[now], mx);
		else del = std::max(fir[now] - dep[now], mx);
		if(mark[edge[i].to]) ans[edge[i].to] = del + fir[edge[i].to] - dep[now];
		calc(edge[i].to, del + 1);
	}
}
void out(R int now)
{
	if(!now) return;
	out(now / 10);
	putchar('0' + now % 10);
}
int main(void)
{
	int T, buf;
	in(dot); in(T);
	W (T--)
	{
		in(dot), in(line);
		std::memset(head, cnt = 0, sizeof(head));
		std::memset(fir, 0, sizeof(fir));
		std::memset(sec, 0, sizeof(sec));
		std::memset(mark, false, sizeof(mark));
		dep[1] = 0;
		for (R int i = 2; i <= dot; ++i) in(buf), add(buf, i);
		for (R int i = 1; i <= line; ++i) in(buf), mark[buf] = true, sta[i] = buf;
		DFS(1);
		calc(1, 0);
		for (R int i = 1; i <= line; ++i) out(ans[sta[i]]), putchar(' ');
		if(line) puts("");
	}
}