【JSOI2016】最佳团队

Description

Input

Output

Sample Input

1 2
1000 1 0
1 1000 1

Sample Output

0.001

Data Constraint

分析

这种最大化形如$\sum x / \sum y$ 的式子的题，很容易想到分数规划。
二分答案，对于当前的mid，设d[i]=p[i]-mid*s[i]
考虑到当一个节点i被选，那么fa[i]也要被选，那么一个想法是选取当前最大的d，然后把它的儿子的d放入堆里。
但是这个方法是错误的（然而我一开始就这么打了，只有10分）

题目给出的是一棵树，那么可以先dfs一遍，存下每个点的dfs序，然后设f[i][j]表示当前做到dfn为i的节点，选取了j个的最大和。设R[i]为节点i的子树中最大的dfn+1
转移有两种情况：
1. 选取i，设它原来的编号为x，那么f[i][j]+a[x]——>f[i+1][j+1]
2. 不选取i，那么f[i][j]——>f[R[i]][j] （这样就跳过了它的子树）

加上二分，时间复杂度为$O(n^2logans)$

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef double db;

const int maxn=2505;

const db z=1e-5,Inf=-1e20;

int m,n,tot,fa[maxn],h[maxn],e[maxn],next[maxn],dfn[maxn],R[maxn];

db d[maxn],s[maxn],p[maxn],f[maxn][maxn],l,r,mid,ans;

void add(int x,int y)
{
    e[++tot]=y; next[tot]=h[x]; h[x]=tot;
}

void init(int x)
{
    dfn[x]=tot++;
    for (int i=h[x];i;i=next[i]) init(e[i]);
    R[dfn[x]]=tot;
}

int main()
{
    scanf("%d%d",&m,&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%lf%lf%d",&s[i],&p[i],&fa[i]);
        add(fa[i],i);
    }
    tot=0;
    init(0);
    for (l=0,r=1e4,mid=r/2;r-l>=z;mid=(l+r)/2)
    {
        for (int i=1;i<=n;i++)
        {
            d[dfn[i]]=p[i]-mid*s[i];
            for (int j=0;j<=m+1;j++) f[i][j]=Inf;
        }
        for (int j=0;j<=m+1;j++) f[n+1][j]=Inf;
        for (int i=0;i<=n;i++)
            for (int j=0;j<=min(i,m+1);j++)
            {
                if (f[i][j]+d[i]>f[i+1][j+1]) f[i+1][j+1]=f[i][j]+d[i];
                if (f[i][j]>f[R[i]][j]) f[R[i]][j]=f[i][j];
            }
        if (f[n+1][m+1]>=0) l=mid;else r=mid;
    }
    printf("%.3lf\n",mid);
    return 0;
}