【vijos1642】班长的任务

最新推荐文章于 2017-09-13 20:06:00 发布

转载最新推荐文章于 2017-09-13 20:06:00 发布 · 74 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/ITUPC/p/5391566.html

本文介绍了一种基于动态规划的算法优化方案，通过记录节点信息实现O(nm)的时间复杂度。文章提供了完整的代码实现，并详细解释了关键步骤，如先序遍历记录根节点及动态转移方程。

思路：这题就是学习一下算法优化，选择最优方案，O（nm）

可以学习一下《浅谈数据的合理组织》这篇论文

详见代码

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<queue>
 5 #include<stack>
 6 #include<bits/stdc++.h>
 7 using namespace std;
 8 #define LL long long
 9 #define clc(a,b) memset(a,b,sizeof(a))
10 #define inf 0x3f3f3f3f
11 const int maxn=1010;
12 //countt数组记录每个节点以下所有子节点的数目，s每个节点下代节点数目 
13 inline int r()
14 {
15     int x=0,f=1;char ch=getchar();
16     while(ch>'9'||ch<'0'){if(ch=='-') f=-1;ch=getchar();}
17     while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
18     return x*f;
19 }
20 int a[maxn][maxn],listt[maxn],countt[maxn],s[maxn],v[maxn];
21 int f[maxn][maxn];
22 int q;
23 int tree_traversal(int k)//先序遍历记录根节点
24 {
25     listt[++q]=k;
26     if(s[k]==0){return ++countt[k];}
27     for(int i=1;i<=s[k];i++)countt[k]+=tree_traversal(a[k][i]);
28     return ++countt[k];
29 }
30 int main(){
31     int n,m;
32     clc(s,0);
33     clc(listt,0);
34     clc(a,0);
35     clc(countt,0);
36     n=r();
37     m=r();
38     int t;
39     for(int i=1;i<=n;i++){
40         v[i]=r(),t=r();
41         while(t--){
42             a[i][++s[i]]=r();
43         }
44     }
45     tree_traversal(1);
46     for(int i=1;i<=n;i++){
47         f[i][0]=0;
48         f[i][1]=v[listt[i]];
49     }
50     for(int i=n-1;i>=1;i--){//动态转移方程
51         for(int j=1;j<=m;j++){
52             f[i][j]=max(f[i+1][j-1]+v[listt[i]],f[i+countt[listt[i]]][j]);
53         }
54     }
55     int ans=-inf;
56     for(int i=1;i<=m;i++){
57         ans=max(ans,f[1][i]);
58     }
59     printf("%d\n",ans);
60     return 0;
61 }