【vijos1642】班长的任务

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

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

详见代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<stack>
#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define clc(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
const int maxn=1010;
//countt数组记录每个节点以下所有子节点的数目，s每个节点下代节点数目 
inline int r()
{
    int x=0,f=1;char ch=getchar();
    while(ch>'9'||ch<'0'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int a[maxn][maxn],listt[maxn],countt[maxn],s[maxn],v[maxn];
int f[maxn][maxn];
int q;
int tree_traversal(int k)//先序遍历记录根节点
{
    listt[++q]=k;
    if(s[k]==0){return ++countt[k];}
    for(int i=1;i<=s[k];i++)countt[k]+=tree_traversal(a[k][i]);
    return ++countt[k];
}
int main(){
    int n,m;
    clc(s,0);
    clc(listt,0);
    clc(a,0);
    clc(countt,0);
    n=r();
    m=r();
    int t;
    for(int i=1;i<=n;i++){
        v[i]=r(),t=r();
        while(t--){
            a[i][++s[i]]=r();
        }
    }
    tree_traversal(1);
    for(int i=1;i<=n;i++){
        f[i][0]=0;
        f[i][1]=v[listt[i]];
    }
    for(int i=n-1;i>=1;i--){//动态转移方程
        for(int j=1;j<=m;j++){
            f[i][j]=max(f[i+1][j-1]+v[listt[i]],f[i+countt[listt[i]]][j]);
        }
    }
    int ans=-inf;
    for(int i=1;i<=m;i++){
        ans=max(ans,f[1][i]);
    }
    printf("%d\n",ans);
    return 0;
}