CF1083A The Fair Nut and the Best Path（树形dp）

给定一棵包含 n 个结点的树，树上的每个结点拥有点权 wi​，每一条边拥有边权 ci​

你可以从树的某个结点出发，沿着树边通往一个未曾经过的结点。每当你到达一个新的结点（包括起点）时，你能获得该结点点权的收益，同时，每经过一条树边，你都会损失该条边边权的收益。

你需要规划出一条路径，使得你在沿着该路径行进的任意时刻收益值均非负，且行进结束时的收益值应尽可能大。输出结束时的最大收益值。

dp[x][0]表示以x作为路径的两端点时的最优解，dp[x][1]表示以x作为路径中点时的最优解

那么由于每个点只能访问一次，中点的dp值就从最大的zhong儿子和次大的儿子转移过来，终点的dp值只能从最大的儿子处转移过来

为了控制路径收益时刻大于0，把最大值和次大值初值设定为0即可

代码：

#include <bits/stdc++.h>
#define int long long
#define sc(a) scanf("%lld",&a)
#define scc(a,b) scanf("%lld %lld",&a,&b)
#define sccc(a,b,c) scanf("%lld %lld %lld",&a,&b,&c)
#define schar(a) scanf("%c",&a)
#define pr(a) printf("%lld",a)
#define fo(i,a,b) for(int i=a;i<b;++i)
#define re(i,a,b) for(int i=a;i<=b;++i)
#define rfo(i,a,b) for(int i=a;i>b;--i)
#define rre(i,a,b) for(int i=a;i>=b;--i)
#define prn() printf("\n")
#define prs() printf(" ")
#define mkp make_pair
#define pii pair<int,int>
#define pub(a) push_back(a)
#define pob() pop_back()
#define puf(a) push_front(a)
#define pof() pop_front()
#define fst first
#define snd second
#define frt front()
#define bak back()
#define mem0(a) memset(a,0,sizeof(a))
#define memmx(a) memset(a,0x3f3f,sizeof(a))
#define memmn(a) memset(a,-0x3f3f,sizeof(a))
#define debug
#define db double
#define yyes cout<<"YES"<<endl;
#define nno cout<<"NO"<<endl;
using namespace std;
typedef vector<int> vei;
typedef vector<pii> vep;
typedef map<int,int> mpii;
typedef map<char,int> mpci;
typedef map<string,int> mpsi;
typedef deque<int> deqi;
typedef deque<char> deqc;
typedef priority_queue<int> mxpq;
typedef priority_queue<int,vector<int>,greater<int> > mnpq;
typedef priority_queue<pii> mxpqii;
typedef priority_queue<pii,vector<pii>,greater<pii> > mnpqii;
const int maxn=500005;
const int inf=0x3f3f3f3f3f3f3f3f;
const int MOD=100000007;
const db eps=1e-10;
int qpow(int a,int b){int tmp=a%MOD,ans=1;while(b){if(b&1){ans*=tmp,ans%=MOD;}tmp*=tmp,tmp%=MOD,b>>=1;}return ans;}
int lowbit(int x){return x&-x;}
int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
int mmax(int a,int b,int c){return max(a,max(b,c));}
int mmin(int a,int b,int c){return min(a,min(b,c));}
void mod(int &a){a+=MOD;a%=MOD;}
bool chk(int now){}
int half(int l,int r){while(l<=r){int m=(l+r)/2;if(chk(m))r=m-1;else l=m+1;}return l;}
int ll(int p){return p<<1;}
int rr(int p){return p<<1|1;}
int mm(int l,int r){return (l+r)/2;}
int lg(int x){if(x==0) return 1;return (int)log2(x)+1;}
bool smleql(db a,db b){if(a<b||fabs(a-b)<=eps)return true;return false;}
db len(db a,db b,db c,db d){return sqrt((a-c)*(a-c)+(b-d)*(b-d));}
bool isp(int x){if(x==1)return false;if(x==2)return true;for(int i=2;i*i<=x;++i)if(x%i==0)return false;return true;}
inline int read()
{
    char ch=getchar();int s=0,w=1;
    while(ch<48||ch>57){if(ch=='-')w=-1;ch=getchar();}
    while(ch>=48&&ch<=57){s=(s<<1)+(s<<3)+ch-48;ch=getchar();}
    return s*w;
}
inline void write(int x)
{
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);
    putchar(x%10+48);
}

int n,w[maxn],x,y,z,ans=0;
vei e[maxn],v[maxn];
int dp[maxn][2];

void dfs(int x,int f){
    int mx1=0,mx2=0;
    fo(i,0,e[x].size()){
        int y=e[x][i];
        if(y==f) continue;
        dfs(y,x);
        if(dp[y][0]-v[x][i]>=mx1){
            mx2=mx1;
            mx1=dp[y][0]-v[x][i];
        }
        else if(dp[y][0]-v[x][i]>mx2)
            mx2=dp[y][0]-v[x][i];
    }
    dp[x][0]=mx1+w[x];
    dp[x][1]=mx1+mx2+w[x];
    ans=mmax(ans,dp[x][0],dp[x][1]);
}

signed main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    cin>>n;
    re(i,1,n) cin>>w[i];
    re(i,1,n-1){
        cin>>x>>y>>z;
        e[x].pub(y);
        v[x].pub(z);
        e[y].pub(x);
        v[y].pub(z);
    }
    dfs(1,1);
    cout<<ans<<endl;
    return 0;
}

 

转载于:https://www.cnblogs.com/oneman233/p/11493222.html