洛谷 P4322 [JSOI2016]最佳团体 分数规划+dp

题目大意：
给定一棵树，$n$个节点。每个节点有$a_i$和$b_i$两个值。要求选出若干节点，保证某个节点被选，他的父亲一定被选。
使
$\frac{{\sum }_{i=1}^k{a}_i}{{\sum }_{i=1}^k{b}_i}$最大化。

$n\le 1000$

分析：
考虑分数规划，二分一个$mid$，
当$mid$合法时，有
$$\frac{{\sum }_{i=1}^k{a}_i}{{\sum }_{i=1}^k{b}_i}\ge mid$$
即
$$\sum _{i=1}^k{a}_i-mid\ast \sum _{i=1}^k{b}_i\ge 0$$
也就是
$$\sum _{i=1}^k{a}_i-mid\ast b_i\ge 0$$
把点的权值设为$a_i-mid\ast b_i$，考虑树上dp。设$f[i][j]$表示$i$的子树中选$j$个的最大值，看是否≥0。
转移要使用枚举$size$ 的方法，dp复杂度是$O(n^2)$的。

代码：

// luogu-judger-enable-o2
#include <iostream>
#include <cstdio>
#include <cmath>

const int maxn=2507;
const double esp=1e-5;

using namespace std;

int n,m,cnt,x;
double l,r,mid,ans;
int a[maxn],b[maxn],size[maxn],ls[maxn];
double f[maxn][maxn],tmp[maxn],val[maxn];

struct edge{
    int y,next;
}g[maxn*2];

void add(int x,int y)
{
    g[++cnt]=(edge){y,ls[x]};
    ls[x]=cnt;
}

void dfs(int x,int fa)
{
    size[x]=1,f[x][0]=0,f[x][1]=val[x];
    for (int i=ls[x];i>0;i=g[i].next)
    {
        int y=g[i].y;
        dfs(y,x);
        for (int j=1;j<=size[x]+size[y];j++) tmp[j]=f[x][j];
        for (int j=1;j<=size[x];j++)
        {
            for (int k=0;k<=size[y];k++)
            {
                tmp[j+k]=max(tmp[j+k],f[x][j]+f[y][k]);
            }
        }
        size[x]+=size[y];
        for (int j=1;j<=size[x];j++) f[x][j]=max(f[x][j],tmp[j]);
    }
}

int main()
{
    scanf("%d%d",&m,&n);
    m++;
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d%d",&b[i],&a[i],&x);
        add(x,i);
    }	
    l=0,r=1e8;
    while (r-l>esp)
    {
        mid=(l+r)/2;
        for (int i=1;i<=n;i++) val[i]=(double)a[i]-b[i]*mid;
        for (int i=0;i<=n;i++)
        {
            for (int j=0;j<=m;j++) f[i][j]=-0x3f3f3f3f;
        }
        dfs(0,0);
        if (f[0][m]>=0) l=mid,ans=mid;
                   else r=mid;
    }
    printf("%.3lf",ans);
}