Gardening Friends

又来写 树 代码了 hh！

实际上，这道题就是我上次写树题解前做的一道类似的题（都是不告诉你父子关系，要自己求）

但这题有多个问题，会更难一些。

题面

题意翻译

有一棵有 n 个节点的树，根节点为节点 1，每条边的权值为 k。现在可以进行挪根操作，每次耗费 c 价值，将树的根转移到与原来根结点相邻的点上。

定义这棵树的价值为根节点到子节点的最远距离 a 与挪根耗费总价值 b 之差（可能不会挪根）。求这棵树经过挪根操作后的最大价值

输入格式

第一行一个正整数 t（1≤t≤10^4）为数据组数。

每组数据的第一行包括整数 n，k，c（2≤n≤2^105，1≤k,c≤10^9），分别为节点数、边权和操作耗费价值。

接下来 n−1 行每行描述一条边，节点为 ui,vi（1≤ui,vi≤n）。

输出格式

一行一个正整数表示最大价值。

输入输出样例

输入 #1复制

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

输出 #1复制

2
12
17
32

洛谷的题目比较直接，只给你满分数据范围。好吧，只能介绍满分做法了。

思路

因为是多组数据，所以我们每次先清空数组，然后读入双向边，存入数组。因为 n<=2*10^5，所以只能用vecor动态分配内存才不会炸。

求距离的数组不需要全部归零，从哪里开始 dfs 就把哪个位置归零就行，因为之后循环到的位置都是由上一个位置更新而来，直接覆盖。

首先从根节点 dfs 一遍，距离数组当然就表示从根节点到当前节点的距离，这也会成为最后挪根的次数，所以要用存一下。

其次找到离根节点最长的距离的编号，再把这个编号当作根节点 dfs 一遍。查找编号可以遍历一遍，当然也可以用函数 max_element。

第二次 dfs 当然就是求距离了，也用一个数组记录一下。

接着找到最远距离的编号，再来一遍 dfs，让前后两者距离求个 max，就是最长距离。

但是最大价值还得跟挪根次数挂钩，有可能距离虽然长，但挪根次数多，不是最大价值。

ans=max(ans,k*w[i]-c*dis[i]);

最后输出 ans 就是最大价值。

时间复杂度 O(t*n)

乘起来 2*10^9 ，不知为何不炸。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=2e5+5;//n等于 2*10^5 !!!
ll T,n,k,c,dep[N],dis[N],w[N];
vector<int>v[N];
void dfs(int u,int fa){
	for(auto s:v[u]){
		if(s!=fa){
			dep[s]=dep[u]+1;//求长度
			dfs(s,u);
		}
	}
}
int main(){
	cin>>T;
	while(T--){
		cin>>n>>k>>c;
		for(int i=1;i<=n;i++)
			vector<int>().swap(v[i]);
		for(int i=1;i<n;i++){
			int a,b;
			cin>>a>>b;
			v[a].push_back(b);
			v[b].push_back(a);
		}
		dep[1]=0;
		dfs(1,0);
		for(int i=1;i<=n;i++)
			dis[i]=dep[i];
		int x=max_element(dep+1,dep+n+1)-dep;//最大值的位置（减去dep才会得到地址）
		dep[x]=0;
		dfs(x,0);
		for(int i=1;i<=n;i++)w[i]=dep[i];//第一次更新
		int y=max_element(dep+1,dep+n+1)-dep;
		dep[y]=0;
		dfs(y,0);
		for(int i=1;i<=n;i++)w[i]=max(w[i],dep[i]);//第二次更新
		ll ans=0;
		for(int i=1;i<=n;i++)//答案取最大值
			ans=max(ans,k*w[i]-c*dis[i]);
        //三次 dfs 结束！
		cout<<ans<<endl;
	}
	return 0;
}

仔细一想，代码应该不属于思路吧！ 

这题是CodeForces的原题，大家有兴趣可以去做一下。