題意:A和B要去旅游。這有N個城市。而這些的城市的道路為一顆樹。且邊有向。A和B從0出發一起去旅游。A很懶想盡量少走路。B很有活力想盡量多走路但是無論怎麼選擇他們走的總路程都必須滿足在[L,R]的范圍內。所以他們交替選擇走哪條路。開始由B選。然後問你在都采用最優策略的情況下。B最多能走多少路。
思路:用dp[i]表示到第i個節點的能獲得最大價值,一道不算難的樹形DP,注意記錄從頭到現在節點所走過的長度,還有就是已經固定了B先選,然後交替選擇所以比較簡單
#include#include #include #include #include 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 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; }