BZOJ 4753 [Jsoi2016]最佳团体 ——01分数规划 树形DP

要求比值最大，当然用分数规划。

二分答案，转化为选取一个最大的联通块使得它们的和大于0

然后我们直接DP。

复杂度$O(n^2log {n})$

#include <map>
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define F(i,j,k) for (int i=j;i<=k;++i)
#define D(i,j,k) for (int i=j;i>=k;--i)
#define ll long long
#define inf 1e15
#define eps 1e-6
#define mp make_pair
#define maxn 2505
int k,n;
double s[maxn],p[maxn],l,r,mid,dp[maxn][maxn],tmp[maxn];
int h[maxn],to[maxn],ne[maxn],en=0,siz[maxn];
 
void add(int a,int b)
{to[en]=b;ne[en]=h[a];h[a]=en++;}
 
void Tree_DP(int o)
{
    dp[o][1]=p[o]-mid*s[o];siz[o]=1;
    for (int i=h[o];i>=0;i=ne[i])
    {
        Tree_DP(to[i]);
        F(j,0,siz[o]+siz[to[i]]) tmp[j]=-inf;
        F(j,1,siz[o]) F(k,0,siz[to[i]])
            tmp[j+k]=max(tmp[j+k],dp[o][j]+dp[to[i]][k]);
        F(j,0,siz[o]+siz[to[i]]) dp[o][j]=max(dp[o][j],tmp[j]);
        siz[o]+=siz[to[i]];
    }
}
 
int main()
{
    memset(h,-1,sizeof h);
    scanf("%d%d",&k,&n);k++;
    F(i,1,n)
    {
        int x;
        scanf("%lf%lf%d",&s[i],&p[i],&x);
        add(x,i);
    }
    l=0;r=1e10;
    while (fabs(r-l)>eps)
    {
        mid=(l+r)/2;
        F(i,0,n) F(j,0,k) dp[i][j]=-inf;
        s[0]=1;p[0]=mid;
        Tree_DP(0);
        int flag=0;
        if (dp[0][k]>=0) l=mid;
        else r=mid;
    }
    printf("%.3f
",l);
}