UVALive - 5088 Alice and Bob's Trip

题意:A和B要去旅游。这有N个城市。而这些的城市的道路为一颗树。且边有向。A和B从0出发一起去旅游。A很懒想尽量少走路。B很有活力想尽量多走路但是无论怎么选择他们走的总路程都必须满足在[L,R]的范围内。所以他们交替选择走哪条路。开始由B选。然后问你在都采用最优策略的情况下。B最多能走多少路。

思路:用dp[i]表示到第i个节点的能获得最大价值,一道不算难的树形DP,注意记录从头到现在节点所走过的长度,还有就是已经固定了B先选,然后交替选择所以比较简单

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAXN = 500050;

struct edge{
    int v,w;
    edge(){}
    edge(int vv,int ww):v(vv),w(ww){}
};
vector<edge> arr[MAXN];
int L,R,n;
int dp[MAXN];

void dfs(int x,int cnt,int pres){
    dp[x] = 0;
    if (arr[x].size() == 0)
        return;
    int temp = 0;
    if (cnt & 1){
        temp = INF;
        for (int i = 0; i < arr[x].size(); i++){
            int y = arr[x][i].v;
            dfs(y,cnt+1,pres+arr[x][i].w);
            if (pres + dp[y] + arr[x][i].w >= L && pres + dp[y] + arr[x][i].w <= R)
                temp = min(temp,dp[y]+arr[x][i].w);
        }
        dp[x] = temp;
    }
    else {
        temp = -INF;
        for (int i = 0; i < arr[x].size(); i++){
            int y = arr[x][i].v;
            dfs(y,cnt+1,pres+arr[x][i].w);
            if (pres + dp[y] + arr[x][i].w >= L && pres + dp[y] + arr[x][i].w <= R)
                temp = max(temp,dp[y]+arr[x][i].w);
            
        }
        dp[x] = temp;
    }
}

int main(){
    while (scanf("%d%d%d",&n,&L,&R) != EOF){
        int x,y,z;
        for (int i = 0; i < n; i++)
            arr[i].clear();
        for (int i = 0; i < n-1; i++){
            scanf("%d%d%d",&x,&y,&z);
            arr[x].push_back(edge(y,z));
        }
        dfs(0,0,0);
        if (dp[0] >= L && dp[0] <= R)
            printf("%d\n",dp[0]);
        else printf("Oh, my god!\n");
    }
    return 0;
}



你可能感兴趣的:(UVALive - 5088 Alice and Bob's Trip)