UVALive - 5088 Alice and Bob's Trip 树形DP

最新推荐文章于 2022-03-22 11:24:55 发布

Peppermint__

最新推荐文章于 2022-03-22 11:24:55 发布

阅读量318

点赞数

CC 4.0 BY-SA版权

分类专栏： DP 文章标签：树形dp

本文链接：https://blog.youkuaiyun.com/Peppermint__/article/details/70228012

DP 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一个树形动态规划的问题实现，通过递归遍历树结构并根据奇偶性选择最大值或最小值来解决问题。代码中详细展示了如何利用C++进行树形DP的设计与实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

第一个树形DP，做题的时候想出来了，但是一直没有做对。什么时候返回的情况下出了问题。看着题解写的。。。

#include <bits/stdc++.h>

using namespace std; 

const int N = 500500; 
const int inf = 0x3f3f3f3f; 

struct node {
	int v,w; 
	node () {}; 
	node ( int vv,int ww ) {v = vv; w = ww; }
};
int n,l,r;
int dp[N]; 

vector<node> edge[N]; 

void dfs( int x,int cnt,int pres)
{	
	dp[x] = 0 ; 
	if ( edge[x].size()==0 ) return ; 
	if ( cnt&1 ) //Min
	{
		int temp = inf; 
		
		for ( int i=0;i<edge[x].size(); i++ ) 
		{
			int to = edge[x][i].v; 
			dfs ( to,cnt+1,pres+edge[x][i].w ) ; 
			if ( pres+edge[x][i].w+dp[to]>=l && pres+edge[x][i].w+dp[to]<=r ) 
			{
				temp = min(temp,edge[x][i].w+dp[to]);
			}
		}
		dp[x] = temp; 
	}
	else   //max
	{
		int temp = -inf; 
		
		for ( int i=0;i<edge[x].size(); i++ ) 
		{
			int to = edge[x][i].v; 
			dfs ( to,cnt+1,pres+edge[x][i].w ) ; 
			if ( pres+edge[x][i].w+dp[to]>=l && pres+edge[x][i].w+dp[to]<=r ) 
			{
				temp = max(temp,edge[x][i].w+dp[to]);
			}
		}
		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++ ) edge[i].clear(); 
		for ( int i=1; i<n; i++ ) 
		{
			scanf("%d%d%d",&x,&y,&z); 
			edge[x].push_back( node(y,z) ) ; 
		}
		dfs( 0,0,0 ); 
		if ( dp[0]>=l && dp[0]<=r ) cout<<dp[0]<<endl; 
		else cout<<"Oh, my god!"<<endl; 
	}
	return 0; 
}