[HDU] 3660 Alice and Bob's Trip -- 树形DP？

哈尔滨现场赛没做出来...

我的代码和思路都参考了这位牛的博客：

http://www.cppblog.com/Yuan/archive/2010/10/03/128074.html?opt=admin

这种做法在我看来就是贪心+暴力么..

我刚开始用vector做，一个一个clear()，超时。然后memset它，MLE。google了一下，知道了C++STL里面的对象不能memset，毕竟里面还有各种成员变量、函数是我们看不到的，很容易造成内存泄漏。

然后用指针做，一需要新的结点就malloc，提交上去又MLE..这时我终于明白上面网址中代码里面的nodes是干什么的了。。真是神奇哈！

提前开好一个数组nodes，每次需要新结点就从nodes里面要，这样就没有浪费内存的问题了！然后改好，提交，通过~

 

Problem : 3660 ( Alice and Bob's Trip )     Judge Status : Accepted
RunId : 3035244    Language : C++    Author : yueashuxia
Code Render Status : Rendered By HDOJ C++ Code Render Version 0.01 Beta

#include<stdio.h> #include<limits.h> #include<algorithm> using namespace std; #define size 500001 struct NODE { int v ; int len ; NODE *next ; }; struct TREE { NODE *first ; }; int dist[size], L, R, dp[size] ; TREE tree[size] ; NODE nodes[size], *first ; void dfs(int u, bool flag) { if(dist[u] > R) {dist[u] = 0; return ;} dp[u] = flag? 0: INT_MAX ; if(tree[u].first == NULL) {dp[u] = 0 ; return ;} for(NODE *list = tree[u].first; list; list = list -> next) { int v = list -> v ; int w = list -> len ; dist[v] = dist[u] + w ; dfs(v, !flag) ; if(dist[u] + dp[v] + w <= R && dist[u] + dp[v] + w >= L) { if(flag) dp[u] = max(dp[u], dp[v] + w) ; else dp[u] = min(dp[u], dp[v] + w) ; } } } int main() { int n, i, j, k, a, b, c; while(scanf("%d%d%d", &n, &L, &R) != EOF) { first = nodes ; for(i = 0; i < n; i ++) tree[i].first = NULL; for(i = 0; i < n-1; i ++) { scanf("%d%d%d", &a, &b, &c); NODE *newnode = first ; newnode -> next = tree[a].first ; newnode -> v = b ; newnode -> len = c ; tree[a].first = newnode ; first ++ ; } dist[0] = 0 ; dfs(0, 1) ; if(dp[0] < L || dp[0] > R) printf("Oh, my god!/n"); else printf("%d/n", dp[0]); } return 0 ; }