UPC2021个人训练赛第39场 C: 粉兔找妹子(换根dp)

问题 C: 粉兔找妹子

时间限制: 1 Sec 内存限制: 128 MB

题目描述
粉兔的妹子跑了！

于是，粉兔要去找她跑掉的妹子！

粉兔的妹子在福州某中上学 。

福州某中共有 N 个路口 ，编号从 1 到 N。路口之间由 N−1 条双向通行的道路连接，并且任意两个路口之间都是连通的。所有只有一条道路与之连接的路口都是福州某中的出入口。一些路口旁有教学楼。粉兔的妹子每天会从某个入口进去，到某栋教学楼上课，然后再从某个出口离开福州某中 。

由于妹子热爱锻炼身♂体， 所以他总是想使自己一天中走的总路程尽量长。当然，妹子不喜欢走重复的路，所以每天他都不会经过同一条路两次。也就是说，妹子会选择一对出入口使得从到教学楼、出口的距离和最大，并且两条路径不会经过同一条道路。可以证明，在不重复经过同一条道路的前提下，连接两个路口的路径是唯一的，从而它们间的距离也是唯一的。

粉兔通过某种手段控制了福州某中所有教学楼里的监摄像头，并掌握了每天妹子上课的教学楼。现在她想知道，妹子每天分别走了多长的路？

注意两个相同路口的距离为 0。
输入
第一行两个整数 N 和 M，分别表示福州某中的路口个数和总天数。
接下来 N−1 行，每行三个整数，Si，Ti和 Li，代表一条连接路口 Si 和 Ti、长度为 Li的道路 。
接下来 M 行，每一个整数 Ai，代表妹子第 i 天上课的教学楼旁路口编号。
输入数据保证合法。
输出
输出 M 行，分别代表妹子每一天走的路程之和。
样例输入 Copy
8 3
1 2 2
2 3 4
3 4 1
4 5 7
4 6 3
2 7 5
1 8 6
3
6
7
样例输出 Copy
20
16
17
提示

对于3号路口，粉兔的妹子会选择从5号入口出发，依次经过路口4,3,2,1走到8号出口（反着走也可以），总长度：7+1+4+2+6=20；
对于6号路口，妹子选择的路径为6→4→3→2→1→8，总长度：3+1+4+2+6=16；
对于7号路口，妹子选择的路径为7→2→3→4→5，总长度：5+4+1+7=17。

对于全部数据， 2≤N≤5×$10^5$，1≤M≤$10^5$，1≤Si,Ti,Ai≤N，1≤Li≤$10^3$。

题意：

给出一棵点数为$5e5$的带权树，$1e5$次询问，每次询问经过$x$节点的最长路径(最长指的是权值和最大)；

思路：

首先要了解用树形$DP$求树的直径的思路，本题跟其思路相近。
用$dp[u][0]$表示从$u$到以$u$的子树里的最长距离；
$dp[u][1]$表示从$u$到以$u$的子树里的次长距离；
根据树形$DP$的$dfs$我们就可以处理出这些。
考虑最长路径跟$dp$数组有什么关系？$dp$数组仅仅是针对子树里的，也就是向下扩展的，而最长路径还可以向上扩展。所以，我们再从根节点$dfs$一遍，求经过每个点的最长路径，最后$O(1)$查询。
首先，根节点的$dp$一定是他本身的答案，因为对于根节点，我们遍历了他所有的节点；而其他节点，只遍历了从根节点到该节点方向的节点；
从根节点向下递推，假设现在正在更新$v$节点，而$v$的父亲节点$u$节点已经被更新过了，那么先要考虑$u$节点的最长路径是否经过$v$节点，如果经过的话，只能用$u$节点的次长路径去更新$v$；不经过的话，用$u$节点的最长路径去更新$v$，这就是换根的过程。
比如样例中的，假设根节点为$1$(无根树可以任意指定一个节点作为根节点)。现在$dp$数组的值已经通过$dfs1$求出来了。

	最长距离	次长距离
1	14	6
2	12	5

从$1$推向$2$，可以发现$dp[2][0]+w=dp[1][0]$，也就是说$1$到子树里的最长路径经过$2$，所以只能用$1$到子树里的最长路径去更新$2$，注意还要加上$1,2$的边的权值。
更新完后$2$的最长距离为$12$,次长距离为$8$，两者相加就是经过$2$的最长路径。

代码：

const int maxn=1e6+7;

vector<PLL>g[maxn];
ll dp[maxn][2],ans[maxn],n,m;

void dfs1(int u,int fa){
	dp[u][0]=dp[u][1]=0;
	for(auto t:g[u]){
		ll nex=t.first,w=t.second;
		if(nex==fa) continue;
		dfs1(nex,u);
		if(dp[u][0]<dp[nex][0]+w){
			dp[u][1]=dp[u][0];
			dp[u][0]=dp[nex][0]+w;
		}
		else if(dp[u][1]<dp[nex][0]+w){
			dp[u][1]=dp[nex][0]+w;
		}
	}
}

void dfs2(int u,int fa){
	for(auto t:g[u]){
		ll v=t.first,w=t.second;
		if(v==fa) continue;
		if(dp[v][0]+w==dp[u][0]){
			ll d=w+dp[u][1];
			if(dp[v][0]<d){
				dp[v][1]=dp[v][0];
				dp[v][0]=d;
			}
			else if(dp[v][1]<d){
				dp[v][1]=d;
			}
		}
		else{
			ll d=w+dp[u][0];
			if(dp[v][0]<d){
				dp[v][1]=dp[v][0];
				dp[v][0]=d;
			}
			else if(dp[v][1]<d){
				dp[v][1]=d;
			}
		}
		ans[v]=dp[v][0]+dp[v][1];
		dfs2(v,u);
	}
}

int main(){
	n=read,m=read;
	rep(i,1,n-1){
		ll u=read,v=read,w=read;
		g[u].push_back({v,w});
		g[v].push_back({u,w});
	}
	dfs1(1,-1);
	ans[1]=dp[1][0]+dp[1][1];
	dfs2(1,-1);
	while(m--){
		int x=read;
		printf("%lld\n",ans[x]);
	}

	
	
	
	
	
	
	return 0;
}