首页 > 代码库 > XDOJ_1070_树状DP
XDOJ_1070_树状DP
http://acm.xidian.edu.cn/problem.php?id=1070
注意j从大到小。
#include<iostream> #include<cstring> #include<cstdio> #include<algorithm> #include<vector> using namespace std; int dp[105][105],n,m; vector<int> v[105]; void dfs(int pos,int pre) { for(int i = 0;i < v[pos].size();i++) { if(v[pos][i] == pre) continue; int x = v[pos][i]; dfs(x,pos); for(int j = m;j > 1;j--) { for(int k = 1;k <= j;k++) { dp[pos][j] = max(dp[pos][j],dp[x][j-k]+dp[pos][k]); } } } } int main() { int T; scanf("%d",&T); while(T--) { scanf("%d%d",&n,&m); memset(dp,0,sizeof(dp)); for(int i = 0;i < n;i++) v[i].clear(); for(int i = 0;i < n;i++) scanf("%d",&dp[i][1]); for(int i = 1;i < n;i++) { int a,b; scanf("%d%d",&a,&b); v[a].push_back(b); v[b].push_back(a); } dfs(0,-1); int ans = 0; for(int i = 0;i < n;i++) ans = max(ans,dp[i][m]); printf("%d\n",ans); } return 0; }
XDOJ_1070_树状DP
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。