P3574 [POI2014]FAR-FarmCraft 树形DP，贪心

题目链接

https://www.luogu.com.cn/problem/P3574

题意

n点树，树根为1，穿越道路花费时间1，点带权。你需要遍历树，每个点会在被遍历到后执行操作，时间为权（你不需要在那里停留），最终当你回到1点后，你也需要开始执行时间为权的操作，问最慢完成的操作的最小值。

思路

树形DP+贪心。
我们定义ans为某点为根子树完成所有操作的最小时间。显然叶子节点是点权，我们考虑非叶子节点。

我们假设i点连接了a,b,c三个子树，我们可以发现当我们为abc规定遍历次序后，就已经确定了答案了，因此，我们只需要确定a，b，c的次序即可。我们首先先考虑下确定次序后如何得出答案，比如我们的次序是a,b,c，那么
$$ans[i]=max（va[i],1+ans[a],1+ans[b]+2+2\ast edge[a],1+ans[c]+2+2\ast edge[b]+2+2\ast edge[c]）$$
稍作解释，va数组为点权，ans数组上文提到过，edge数组存放的是以它为根子树边数量。率先到达i点，所以i可以直接开始，需要取max，之后是a，b，c三棵树。a树率先开始，他的时间应该是到达时间+花费时间。花费时间显然是我们定义的ans，到达时间很简单，a第一个访问应当是1.

到了b树，仍然是花费时间ans[b]，那么到达他的时间应该是到达a树时间+a树遍历时间+返回i时间+到达b时间，其中到达a和返回i之和是2，到达b是1，遍历a树的时间就是a树中边个数*2了。

到了c树，是类似的，这时到达时间算上a和b就可以了。

依次类推，我们可以确定多个节点时，确定下遍历次序后答案的值，我们要让他最小，可以使用扰动法。我们写出次序第i个和第j个的式子，我们尝试更换i和j的顺序，整理发现仅当
$$ans[i]-2\ast edge[i]>ans[j]-2\ast edge[j]$$
时，交换i和j可以使答案更优。所以我们排一下序贪心地选就好了。

注意一点就是作为根节点，你必须回到根才可以进行任务，所有最后答案应该是ans[1]和va[1]+2*n-2取一个max

说一下代码实现细节，这个排序的ans值是在DP过程中不断变化的，所以不能提前出边排序，而是应该放到dfs过程中进行，这里也算是get了一个新姿势吧，之前写的dfs都是在for里写回溯部分，这个则是先for一遍跑dfs，回溯时再排序，重新遍历一遍处理，如果没写过类似的还真的会被困扰一下。

代码

#include<cstdio>
#include<iostream>
#include<iomanip>
#include<map>
#include<unordered_map>
#include<string>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib> 
#include<chrono>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl "\n"
#define int long long
//#define double long double
using namespace std;
	typedef long long ll;
	const int maxn=500505;
	const int inf=0x3f3f3f3f;
	int n,m,k;
    int va[maxn];
    vector<int>e[maxn];
    int ans[maxn],edge[maxn];
    bool cmp(int a,int b){
        return ans[a]-2*edge[a]>ans[b]-2*edge[b];
    }
    void cnt_edge(int x,int fa){
        for(auto y:e[x]){
            if(y==fa)   continue;
            cnt_edge(y,x);
            
        }
    }
    void dfs(int x,int fa){
        int time=0;
        ans[x]=va[x];
        for(auto y:e[x]){
            if(y==fa)   continue;
            dfs(y,x);
            edge[x]+=edge[y]+1;
        }
        sort(e[x].begin(),e[x].end(),cmp);
        for(auto y:e[x]){
            if(y==fa)   continue;
            ans[x]=max(ans[x],time+1+ans[y]);
            time+=2*edge[y]+2;
        }
    }
	signed main(){
        IOS
		#ifndef ONLINE_JUDGE
		    freopen("IO\\in.txt","r",stdin);
		    freopen("IO\\out.txt","w",stdout);
        #endif
		int tn=1;
        cin>>n;
        for(int i=1;i<=n;i++)   cin>>va[i];
        for(int i=1;i<n;i++){
            int u,v;
            cin>>u>>v;
            e[u].push_back(v);
            e[v].push_back(u);
        }
        dfs(1,-1);
        cout<<max(ans[1],va[1]+2*n-2)<<endl;
	}