luogu2014 选课 背包类树形DP

题目大意:有N门功课,每门课有个学分,每门课有一门或没有直接先修课(若课程a是课程b的先修课即只有学完了课程a,才能学习课程b)。一个学生要从这些课程里选择M门课程学习,问他能获得的最大学分是多少?

对于每个节点u,定义u->DP[j]为在剩余课程数为j,以u为根的子树中能够获得的最高学分,则u->DP[j]=max{sum{v->DP[k] | sum[k]==j-1}|v∈u->Son}。

当u的子节点的子节点的DP值通过递归得出后,我们可以对u的每个DP[j]做一个背包。sum[k]==j-1提醒我们背包体积为j-1,k为物体的容量。DP求和提示我们DP[k]作为物体的价值。因为凑成j-1时每个v只选出一个v->DP[k],这提示我们背包按照子节点进行分组,物体最多选一个。这便是一个分组背包!之后将u->DP[j]=u->DP[j-1]+u->score.

注意:

  • 分组背包的初值设置不要忘,DP[0]=0,其余为负无穷。
  • 枚举k时要倒序循环,因为要考虑到物品的体积可能为0,如果正序循环,0最后处理,此时的DP[j]已经是更改后的值。
  • DP[j]=DP[j-1]+score,而不是DP[j]=DP[j]+score
    #include <cstdio>
    #include <cstring>
    #include <vector>
    #include <algorithm>
    #include <cstdarg>
    using namespace std;
    
    const int MAX_NODE = 310, MAX_V = MAX_NODE;
    int totV, totNode;
    
    struct Node
    {
    	int DP[MAX_V];
    	int Score;
    	int Id;
    	vector<Node*> Next;
    }_nodes[MAX_NODE];
    
    void AddEdge(int uId, int vId, int vScore)
    {
    	Node *u = uId + _nodes, *v = vId + _nodes;
    	u->Id = uId;
    	v->Id = vId;
    	v->Score = vScore;
    	u->Next.push_back(v);
    }
    
    void Dfs(Node *cur)
    {
    	memset(cur->DP, 0xcf, sizeof(cur->DP));
    	cur->DP[0] = 0;
    	int sonCnt = cur->Next.size();
    	for (int i = 0; i < sonCnt; i++)
    		Dfs(cur->Next[i]);
    	for (int son = 0; son < sonCnt; son++)
    		for (int j = totV; j >= 0; j--)
    			for (int k = j; k >= 0; k--)
    				cur->DP[j] = max(cur->DP[j], cur->DP[j - k] + cur->Next[son]->DP[k]);
    	if(cur!=_nodes)
    		for (int j = totV; j > 0; j--)
    			cur->DP[j] = cur->DP[j - 1] + cur->Score;
    }
    
    int Proceed()
    {
    	Node *root = _nodes;
    	Dfs(root);
    	return root->DP[totV];
    }
    
    int main()
    {
    #ifdef _DEBUG
    	freopen("c:\\noi\\source\\input.txt", "r", stdin);
    #endif
    	memset(_nodes, 0, sizeof(_nodes));
    	int u, score;
    	scanf("%d%d", &totNode, &totV);
    	for (int i = 1; i <= totNode; i++)
    	{
    		scanf("%d%d", &u, &score);
    		AddEdge(u, i, score);
    	}
    	printf("%d\n", Proceed());
    	return 0;
    }
    

      

posted @ 2018-03-11 16:13  headboy2002  阅读(175)  评论(0编辑  收藏  举报