题目的介绍以及思路完全参考了下面的博客:http://blog.csdn.net/acm_cxlove/article/details/7964739
做这道题主要是为了加强自己对SPFA的代码的训练以及对树dp的一些思路的锻炼。我特地研究了一下树dp的部分
1
2
3
4
5 |
for
( int i = t; i >= w; i--){ for
( int j = i-w; j >= 0; j--){ dp[u][i] = max(dp[u][i], dp[u][j]+dp[v][i - j - w]); } } |
循环里面是不能搞错顺序的,外层的i逆序显然,但为什么里层会有问题呢? 因为w是可以=0的,这个时候想像一下,如果j=i-w的话,那么就会有
dp[u][i]=max(dp[u][i],dp[u][i]+dp[v][i-j-w]),但是dp[u][i]是已经更新了的,所以这样会出错,两层循环都必须要逆序。
下面贴一记代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108 |
#pragma warning(disable:4996) #include<iostream> #include<cstring> #include<string> #include<cstdio> #include<algorithm> #include<cmath> #include<vector> #include<queue> #define maxn 150 using
namespace std; struct
Edge { int
u, v, w; Edge(){} Edge( int
ui, int
vi, int
wi) :u(ui), v(vi), w(wi){} }e[2*maxn]; int
n, t; int
val[maxn]; int
ecnt; int
first[maxn], nxt[2 * maxn]; void
add( int
u, int
v, int
w) { e[ecnt].u = u; e[ecnt].v = v; e[ecnt].w = w; nxt[ecnt] = first[u]; first[u] = ecnt++; } int
dis[maxn], vis[maxn]; int
p[maxn]; void
spfa( int
s) { memset (dis, 0x3f, sizeof (dis)); memset (vis, 0, sizeof (vis)); memset (p, -1, sizeof (p)); queue< int > que; que.push(s); vis[s] = 1; dis[s] = 0; while
(!que.empty()) { int
u = que.front(); que.pop(); vis[s] = 0; for
( int i = first[u]; i != -1; i = nxt[i]){ int
v = e[i].v; if
(dis[v] > dis[u] + e[i].w){ dis[v] = dis[u] + e[i].w; p[v] = i; if
(!vis[v]){ que.push(v); vis[v] = 1; } } } } } int
dp[maxn][550]; void
dfs( int
u, int
fa) { for
( int i = first[u]; i != -1; i = nxt[i]){ int
v = e[i].v, w = e[i].w * 2; if
(v == fa) continue ; dfs(v, u); for
( int i = t; i >= w; i--){ for
( int j = i-w; j >= 0; j--){ dp[u][i] = max(dp[u][i], dp[u][j]+dp[v][i - j - w]); } } } for
( int i = 0; i <= t; i++){ dp[u][i] += val[u]; } } int
main() { while
(cin >> n >> t) { int
ui, vi, wi; memset (first, -1, sizeof (first)); ecnt = 0; for
( int i = 0; i < n - 1; i++){ scanf ( "%d%d%d" , &ui, &vi, &wi); add(ui, vi, wi); add(vi, ui, wi); } for
( int i = 1; i <= n; i++) scanf ( "%d" , val + i); spfa(1); if
(dis[n]>t) { puts ( "Human beings die in pursuit of wealth, and birds die in pursuit of food!" ); continue ; } for
( int i = n; i != 1; i = e[p[i]].u){ e[p[i]].w = e[p[i] ^ 1].w = 0; } t -= dis[n]; memset (dp, 0, sizeof (dp)); dfs(1, -1); printf ( "%d\n" , dp[1][t]); } return
0; } |
HDU4276 The Ghost Blows Light SPFA&&树dp
原文:http://www.cnblogs.com/chanme/p/3567172.html