1. 程式人生 > >邱老師玩遊戲 UESTC - 1136

邱老師玩遊戲 UESTC - 1136

%d 空間 closed ide 特定 位置 opened ret code

邱老師最近在玩一種戰略遊戲,在一個地圖上,有N座城堡,每座城堡都有一定的寶物,在每次遊戲中邱老師允許攻克M個城堡並獲得裏面的寶物。

但由於地理位置原因,有些城堡不能直接攻克,要攻克這些城堡必須先攻克其他某一個特定的城堡。你能幫邱老師算出要獲得盡量多的寶物應該攻克哪M個城堡嗎?

Input
每個測試實例首先包括2個整數,N,M.(1 <= M <= N <= 200);

在接下來的N行裏,每行包括2個整數,a,b.

在第 i 行,a 代表要攻克第 i 個城堡必須先攻克第 a 個城堡,如果 a = 0 則代表可以直接攻克第 i 個城堡。b 代表第 i 個城堡的寶物數量, b >= 0。

當N = 0, M = 0輸入結束。

Output
對於每個測試實例,輸出一個整數,代表邱老師攻克M個城堡所獲得的最多寶物的數量。

Sample Input
3 2
0 1
0 2
0 3
7 4
2 2
0 1
0 4
2 1
7 1
7 6
2 2
0 0
Sample Output
5
13

//第一次做樹形DP還是比較有意思的,dp[i][j] 代表在i 節點選 j 個的最大收益,滾動數組省空間,還有就是 m++ 很關鍵,因為0節點算進去了

技術分享
 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 #define MX 205
 4 
 5 int
n,m; 6 int dp[MX][MX]; 7 int b[MX]; 8 int vis[MX]; 9 vector<int> tree[MX]; 10 11 void dfs(int x) 12 { 13 if (vis[x])return ; 14 vis[x]=1; 15 dp[x][1]=b[x]; 16 for (int i=0;i<tree[x].size();i++) 17 { 18 int v = tree[x][i]; 19 dfs(v); 20 for
(int j=m;j>=0;j--) 21 { 22 for (int k=0;k<j;k++) 23 dp[x][j] = max(dp[x][j],dp[v][k]+dp[x][j-k]); 24 } 25 } 26 } 27 28 int main() 29 { 30 while (scanf("%d%d",&n,&m)!=EOF) 31 { 32 if (n==0&&m==0) break; 33 m++; 34 for (int i=0;i<=n;i++) tree[i].clear(); 35 memset(vis,0,sizeof(vis)); 36 memset(dp,0,sizeof(dp)); 37 for (int i=1;i<=n;i++) 38 { 39 int a; 40 scanf("%d%d",&a,&b[i]); 41 tree[a].push_back(i); 42 } 43 dfs(0); 44 printf("%d\n",dp[0][m]); 45 } 46 return 0; 47 }
View Code

邱老師玩遊戲 UESTC - 1136