bzoj4753: [Jsoi2016]最佳团体

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=4735

题解

　　在萌神的指点下学会了这道题。
　　先01分数规划，二分答案，
　　

$$\frac{\sum S}{\sum P}\leq ans$$
　　即 $$\sum S-\sum P\times ans\geq 0$$
　　那就重新计算点权 $w_i=s_i-mid\times p_i$，转成判定性问题：能否选择一些点，使得总点权不小于0？
　　传统的泛化背包或者多叉转二叉是 $O(n^3)$的。
　　 $According\ to\ ftm2001$(萌神)，DFS之后，在dfs序的序列上搞。
　　 $f_{ij}$表示在 $dfs$序序列上做了前 $i$个点，已经选择了$j$个人的最大权值和。
　　那么如果这个点选 $f_{ij}->f_{i+1,j+1}$
　　如果不选 $f_{ij}->f_{i+size_i,j}$（表示跳过子树转移）
　　对于这个点选，子树都不选的情况，能够通过第一个方程包含到。

代码

//01分数规划+树形dp
#include <cstdio>
#include <algorithm>
#include <cmath>
#define maxn 3000
#define eps 1e-8
using namespace std;
int tid[maxn], untid[maxn], head[maxn], to[maxn], nex[maxn], tot, K, N, size[maxn];
double P[maxn], S[maxn], w[maxn], f[maxn][maxn], mid;
inline void adde(int a, int b){to[++tot]=b;nex[tot]=head[a];head[a]=tot;}
void dfs(int pos)
{
    int p;
    tid[pos]=++tid[0];
    w[tid[pos]]=P[pos]-mid*S[pos];
    for(p=head[pos];p;p=nex[p])dfs(to[p]);
    size[tid[pos]]=tid[0]-tid[pos]+1;
}
void init()
{
    int r, i;
    scanf("%d%d",&K,&N);
    for(i=1;i<=N;i++)scanf("%lf%lf%d",S+i,P+i,&r),adde(r?r:N+1,i);
}
bool dp()
{
    int i, j;
    tid[0]=0;dfs(N+1);
    for(i=1;i<=N+2;i++)for(j=0;j<=K;j++)f[i][j]=-1e60;
    f[1][0]=0;
    for(i=1;i<=N+1;i++)
        for(j=0;j<=K;j++)
        {
            f[i+1][j+1]=max(f[i+1][j+1],f[i][j]+w[i]);
            f[i+size[i]][j]=max(f[i][j],f[i+size[i]][j]);
        }
    return f[N+2][K]>-eps;
}
int main()
{
    double l, r;
    init();K++;
    for(l=1e-4, r=1e4;fabs(l-r)>1e-5;)
    {
        mid=(l+r)/2.0;
        if(dp())l=mid;else r=mid;
    }
    printf("%.3lf\n",l);
    return 0;
}