POI2014FarmCraft

大意:

就是给你一棵树,然后树上有n个节点,每条边的边权为1,然后每个节点有一个延时t,当你走过这个节点后,过了t时间之后,这个节点就被加入到已经经过的点的集合中,然后从1节点出发,在1节点结束,且1节点的计时只会在最后一次到1节点的时候开始.

求最少时间,能够经过所有的点.

这道题的话,感觉就是一个状态定义出来后就只剩下码代码的树形dp了吧…

那就开始定义状态.

我们定义dp[i]为从第i个点出来之后,里面的点还需要多久才能够结束所有的延时.

那么对于一个节点p,我们可以知道它的延时应该是它的子树中最早开始的,然后我们又可以很快地算出走完这棵子树上所有的边需要多少时间.

然后我们进行一个贪心,对于它的儿子,我们优先走那些dp值大的儿子,这样能够使得它们的能够更早地结束延时,从而使得答案更优.

就是这样,实现的话,在里面开一个vector来排个序就行了.

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#define M 500005
#define LL long long
using namespace std;
void Rd(int &res){
    res=0;char p;
    while(p=getchar(),p<'0');
    do{
        res=(res<<1)+(res<<3)+(p^48);
    }while(p=getchar(),p>='0');
}
struct W{
    int to,nx;
}Lis[M<<1];
int Head[M],tot=0;
void Add(int x,int y){
    Lis[tot].to=y,Lis[tot].nx=Head[x],Head[x]=tot++;
    Lis[tot].to=x,Lis[tot].nx=Head[y],Head[y]=tot++;
}
int n,C[M],cntson[M];
LL dp[M];
int dfs(int x,int f){
    vector<pair<int,int> >son;
    cntson[x]=0;
    for(int i=Head[x];~i;i=Lis[i].nx){
        int to=Lis[i].to;
        if(to==f)continue;
        int dl=dfs(to,x)-1;
        son.push_back(make_pair(dl,cntson[to]*2LL));
        cntson[x]+=cntson[to];
    }
    sort(son.begin(),son.end());
    for(int i=son.size()-1;i>=0;i--){
        pair<int,int>it=son[i];
        int ll=it.first,res=it.second;
        dp[x]-=res;
        if(ll>dp[x])dp[x]=ll;
    }
    LL ll=C[x]-cntson[x]*2LL;
    if(ll>dp[x])dp[x]=ll;
    cntson[x]++;
    if(dp[x]<0)dp[x]=0;
//  printf("x:%d %lld\n",x,dp[x]);
    return dp[x];
}
int main(){
    memset(Head,-1,sizeof(Head));
    Rd(n);
    for(int i=1;i<=n;i++)Rd(C[i]);
    for(int i=1;i<n;i++){
        int x,y;Rd(x),Rd(y);
        Add(x,y);
    }
    cout<<max(dfs(1,0)+(n-1)*2LL,(n-1)*2LL+C[1])<<endl;
    return 0;
}