码迷,mamicode.com
首页 > Web开发 > 详细

【bzoj4753】[Jsoi2016]最佳团体 分数规划+树形背包dp

时间:2017-06-17 18:21:39      阅读:224      评论:0      收藏:0      [点我收藏+]

标签:add   span   define   背包   std   void   集合   个性   使用   

题目描述

JSOI信息学代表队一共有N名候选人,这些候选人从1到N编号。方便起见,JYY的编号是0号。每个候选人都由一位编号比他小的候选人Ri推荐。如果Ri=0则说明这个候选人是JYY自己看上的。为了保证团队的和谐,JYY需要保证,如果招募了候选人i,那么候选人Ri"也一定需要在团队中。当然了,JYY自己总是在团队里的。每一个候选人都有一个战斗值Pi",也有一个招募费用Si"。JYY希望招募K个候选人(JYY自己不算),组成一个性价比最高的团队。也就是,这K个被JYY选择的候选人的总战斗值与总招募总费用的比值最大。

输入

输入一行包含两个正整数K和N。
接下来N行,其中第i行包含3个整数Si,Pi,Ri表示候选人i的招募费用,战斗值和推荐人编号。
对于100%的数据满足1≤K≤N≤2500,0<"Si,Pi"≤10^4,0≤Ri<i

输出

输出一行一个实数,表示最佳比值。答案保留三位小数。

样例输入

1 2
1000 1 0
1 1000 1

样例输出

0.001


题解

分数规划+树形背包dp

二分答案mid,题目便转化为求是否存在满足题目条件的集合V,使得$\frac{\sum\limits_{i\in V}p_i}{\sum\limits_{i\in V}s_i}\ge mid$,即$\sum\limits_{i\in V}(s_i-mid·p_i)\ge 0$。

这就转化为了一个树形dp问题。

令a[i]=s[i]-mid*p[i],表示i的性价比。设f[i][j]表示从子树i中选出j个且选i,可以获得的最大性价比之和,显然f[i][1]=a[i]。

那么对于每个i的子节点son,相当于有体积为1~si[son]共si[son]个物品放入背包内,每个物品可以放或不放。这相当于01背包问题。

但是这样dp的时间复杂度好像是$O(n^3)$的。

事实上,这里面的有效状态是很少的,如果只枚举有效状态,dp的时间复杂度将到达可以接受的$O(n^2)$。

具体粗略证明:

更新一棵子树的时间复杂度=更新该节点的子节点的时间复杂度+计算该节点的时间复杂度。

计算该节点的复杂度,如果采用最优策略,使用严格的有效区间范围来进行dp,时间复杂度应该为

$O(\sum\limits_{i=1}^m(1+\sum\limits_{j=1}^{i-1}si_j)·si_i)=O(\sum\limits_{i=1}^m\sum\limits_{j=1}^{i-1}si_j·si_i+\sum\limits_{i=1}^msi_i)=O((\sum\limits_{i=1}^msi_i)^2-\sum\limits_{i=1}^msi_i^2+\sum\limits_{i=1}^msi_i)=O(si_x^2-\sum\limits_{i=1}^msi_i^2+si_x)$,

其中$si_i(i\in[1,m])$表示x的第i个儿子节点的子树大小(总共有m个儿子节点),$si_x$表示x的子树大小。

而叶子节点的时间复杂度是$O(1)$的,进而我们可以使用累加法计算出总体dp的时间复杂度为$O(si_{root}^2+\sum\limits_{i=1}^nsi_i^2)=O(n^2)$。

因此总的时间复杂度是$O(n^2\log n)$。

为了避免精度误差带来的答案错误,建议固定二分c次,c值视情况而定,本题中取30可过。

另外由于数据太水了,所以$O(n^3\log n)$的做法也是可以通过本题的。(其实可以自己做一个链的数据卡掉它)

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 2510
using namespace std;
int head[N] , to[N] , next[N] , cnt , si[N] , w[N] , v[N] , n;
double a[N] , mid , f[N][N];
void add(int x , int y)
{
	to[++cnt] = y , next[cnt] = head[x] , head[x] = cnt;
}
void init(int x)
{
	int i;
	si[x] = 1;
	for(i = head[x] ; i ; i = next[i]) init(to[i]) , si[x] += si[to[i]];
}
void dfs(int x)
{
	int i , j , k , tot = 0 , b = 0;
	memset(f[x] , 0xc2 , sizeof(f[x]));
	if(x) f[x][1] = a[x] , tot ++ , b ++ ;
	else f[x][0] = 0;
	for(i = head[x] ; i ; i = next[i])
	{
		dfs(to[i]);
		for(j = tot ; j >= b ; j -- )
			for(k = 1 ; k <= si[to[i]] ; k ++ )
				f[x][j + k] = max(f[x][j + k] , f[x][j] + f[to[i]][k]);
		tot += si[to[i]];
	}
}
int main()
{
	int n , k , i , x , c = 30;
	double l = 0 , r = 0;
	scanf("%d%d" , &k , &n);
	for(i = 1 ; i <= n ; i ++ ) scanf("%d%d%d" , &w[i] , &v[i] , &x) , add(x , i) , r = max(r , (double)v[i]);
	init(0);
	while(c -- )
	{
		mid = (l + r) / 2;
		for(i = 1 ; i <= n ; i ++ ) a[i] = v[i] - mid * w[i];
		dfs(0);
		if(f[0][k] >= 0) l = mid;
		else r = mid;
	}
	printf("%.3lf\n" , (l + r) / 2);
	return 0;
}

 

 

【bzoj4753】[Jsoi2016]最佳团体 分数规划+树形背包dp

标签:add   span   define   背包   std   void   集合   个性   使用   

原文地址:http://www.cnblogs.com/GXZlegend/p/7040678.html

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