P4322 [JSOI2016]最佳团体

P4322 [JSOI2016]最佳团体

$01$分数规划与树形$dp$结合

题意：
求$\frac{\sum p_i}{\sum s_i}$的最大值，我们可以二分一个$v$，则我们只需要求$\sum p_i>\sum s_i\ast v==>\sum p_i-\sum s_i\ast v>0$就$ok$

让每一个点的权值等于$p_i-s_i\ast v$，然后做一个树形DP。

树形$dp$就直接用有依赖的背包问题则可以解决

最后一个注意点就是这题有一个虚根，也就是0号节点

Code

#include<bits/stdc++.h>
#define rep(i,a,b) for(int (i)=(a);(i)<=(b);(i)++)
#define ll long long
#define eps 1e-4
using namespace std;
const int maxn=1e5+10;
const int maxm=2500+10;
int n,cnt,m;
double f[maxm][maxm],head[maxn],w[maxn];
int fa[maxn],size[maxn];
double l=0.0,r=10000.0,ans=0;

template <class t> inline void read(t &x) {
	int f=1;x=0;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-') ch=getchar();ch=getchar();}
	while(isdigit(ch)){x=10*x+ch-'0';ch=getchar();}
	x*=f;
}

struct node{
	double cost,v;
}a[maxn];

struct {
	int v,nex;
}e[maxn];

void add(int u,int v) {
	e[++cnt].v=v;
	e[cnt].nex=head[u];
	head[u]=cnt;
}

void readdata() {
	read(m);read(n);
	m++;n++;
	rep(i,2,n) {
		int fa;
		scanf("%lf%lf",&a[i].cost,&a[i].v);
		read(fa);fa++;
		add(fa,i);
	}
}

void dfs(int u,double mid) {
	f[u][0]=0;f[u][1]=a[u].v-mid*a[u].cost;
	size[u]=1;
	for(int i=head[u];i;i=e[i].nex) {
		int v=e[i].v;
		dfs(v,mid);
		size[u]+=size[v];
		for(int j=size[u];j>=0;j--) {
			for(int k=0;k<=size[v] && k<j;k++) {
				f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]);
			}
		}
	}
}

void work() {
	while(l+eps<=r) {
		double mid=(l+r)/2.0;
		//rep(i,1,n) w[i]=a[i].v-mid*a[i].cost;
		memset(f,-0x3f3f,sizeof(f));
	//	memset(size,false,sizeof(size));
		dfs(1,mid);
		if(f[1][m]>=0) l=mid;
		else r=mid;
	}
	printf("%.3lf",l);
}

int main()  {
	//freopen("input.txt","r",stdin);
	readdata();
	work();
	return 0;
}