hdu3660 Alice and Bob's Trip 树形dp

来源：互联网发布：mac输入法表情符号编辑：程序博客网时间：2024/04/28 05:56

dfs传三个参数，一个是当前截点，一个是当前路程和，一个是选择的类型。

当这个点不能选时就不再考虑这个点的值。

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>#include <cmath>using namespace std;#define maxn 1100000#define ll long longint n,L,R,num;int dp[maxn],vis[maxn];int val[maxn],next[maxn],first[maxn],b[maxn];void build(int x,int y,int z){    num++;    b[num]=y;    next[num]=first[x];    first[x]=num;    val[num]=z;}void dfs(int u,int from,int dep){    vis[u]=1;    int v,tmp,maxv=0,flag;    if(!first[u])    {        dp[u]=0;        return;    }    if(dep==0)        dp[u]=-1;    else        dp[u]=0x3f3f3f3f;    for(int i=first[u];i;i=next[i])    {        v=b[i];        if(vis[v]) continue;        dfs(v,from+val[i],dep^1);        if(dp[v]==-1||dp[v]==0x3f3f3f3f)        {            continue;        }        tmp=from+val[i]+dp[v];        if(tmp>=L&&tmp<=R)        {            if(dep)            {                dp[u]=min(dp[u],tmp-from);            }            else            {                dp[u]=max(dp[u],tmp-from);            }        }    }}void solve(){    dfs(0,0,0);}void init(){    memset(first,0,sizeof(first));    memset(val,0,sizeof(val));    memset(dp,0,sizeof(dp));    memset(vis,0,sizeof(vis));    num=0;}int main(){    int a,b,c;    while(~scanf("%d%d%d",&n,&L,&R))    {        init();        for(int i=1;i<n;i++)        {            scanf("%d%d%d",&a,&b,&c);            build(a,b,c);        }        solve();        if(dp[0]>=L&&dp[0]<=R)        {            printf("%d\n",dp[0]);        }        else        printf("Oh, my god!\n");    }    return 0;}

0 0