码迷,mamicode.com
首页 > 其他好文 > 详细

洛谷P2014——选课

时间:2018-02-12 11:28:08      阅读:154      评论:0      收藏:0      [点我收藏+]

标签:gpo   tar   sharp   target   min   blank   ret   www.   i++   

题目:https://www.luogu.org/problemnew/show/P2014

树状DP,注意枚举当前子树中选几个时的边界。

代码如下:

#include<iostream>
#include<cstdio>
using namespace std;
int n,m,k,s[305],ct,cnt[305],dp[305][305],size[305];
struct N{
	int to,next;
}edge[305];
void add(int x,int y)
{
	ct++;
	edge[ct].to=y;
	edge[ct].next=cnt[x];
	cnt[x]=ct;
}
void f(int nw)
{
	for(int i=cnt[nw];i;i=edge[i].next)
	{
		int u=edge[i].to;
		f(u);
		for(int j=min(size[nw]+size[u],m-1);j>=0;j--)//
		{
			for(int k=max(0,j-size[nw]);k<j&&k<=size[u];k++)//
				dp[nw][j]=max(dp[nw][j],dp[nw][j-k]+dp[u][k]);
		}
		size[nw]+=size[u];
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d%d",&k,&s[i]);
		add(k,i);
	}
	for(int i=1;i<=n;i++)size[i]=1,dp[i][1]=s[i];
	for(int i=cnt[0];i;i=edge[i].next)
	{
		int u=edge[i].to;
		f(u);
		for(int j=min(size[0]+size[u],m);j>=0;j--)
		{
			for(int k=max(0,j-size[0]);k<=size[u]&&k<=j;k++)
				dp[0][j]=max(dp[0][j],dp[0][j-k]+dp[u][k]);
		}
		size[0]+=size[u];
	}
	printf("%d",dp[0][m]);
	return 0;
}

 

洛谷P2014——选课

标签:gpo   tar   sharp   target   min   blank   ret   www.   i++   

原文地址:https://www.cnblogs.com/Zinn/p/8443685.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!