作者:小王儿 | 来源:互联网 | 2023-12-09 10:16
本文介绍了UVA1484题目的解题思路和代码实现,题目要求根据给定的条件,计算出使得路径权值总和在指定范围内的最终路径权值。文章详细解释了树形DP的思路和状态转移方程,并提供了代码实现。
题目链接:1484 - Alice and Bob‘s Trip
题意:BOB和ALICE这对狗男女在一颗树上走,BOB先走,BOB要尽量使得总路径权和大,ALICE要小,但是有个条件,就是路径权值总和必须在[L,R]之间,求最终这条路径的权值。
思路:树形dp,dp[u]表示在u结点的权值,往下dfs的时候顺带记录下到根节点的权值总和,然后如果dp[v] + w + sum 在[l,r]内,就是可以的,状态转移方程为
dp[u] = max{dp[v] + w }(bob) dp[u] = min{dp[u] + w} (alice)。所以如果是bob初始化为0,alice初始化为INF。
但是注意如果搜到叶子节点的时候,不管到谁dp[u]都出初始化为0,被这个坑了
还有就是HDU上这题用vector是过不了的,要用数组模拟的链表
代码:
#include
#include
#include
using namespace std;
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define INF 0x3f3f3f3f
const int N = 500005;
int n, l, r, dp[N], E, first[N], next[N];
struct Edge {
int u, v, w;
} edge[N];
inline void scanf_(int &num)//无负数
{
char in;
while((in=getchar()) > ‘9‘ || in<‘0‘) ;
num=in-‘0‘;
while(in=getchar(),in>=‘0‘&&in<=‘9‘)
num*=10,num+=in-‘0‘;
}
void dfs(int u, int fa, int sum, int who) {
if (who && first[u] != -1) dp[u] = INF;
else dp[u] = 0;
for (int i = first[u]; i != -1; i = next[i]) {
int v = edge[i].v, w = edge[i].w;
if (v == fa) continue;
dfs(v, u, sum + w, 1 - who);
if (who == 0 && dp[v] + w + sum >= l && dp[v] + w + sum <= r)
dp[u] = max(dp[u], dp[v] + w);
if (who == 1 && dp[v] + w + sum >= l && dp[v] + w + sum <= r)
dp[u] = min(dp[u], dp[v] + w);
}
}
void add(int u, int v, int w) {
edge[E].u = u; edge[E].v = v; edge[E].w = w;
next[E] = first[u];
first[u] = E++;
}
int main() {
while (~scanf("%d%d%d", &n, &l, &r)) {
E = 0;
memset(first, -1, sizeof(first));
int u, v, w;
for (int i = 0; i r) printf("Oh, my god!\n");
else printf("%d\n", dp[0]);
}
return 0;
}
UVA 1484 - Alice and Bob&#39;s Trip(树形DP),编程笔记,mamicode.com
UVA 1484 - Alice and Bob&#39;s Trip(树形DP)