[二分+DFS序上DP]BZOJ 4753—— [Jsoi2016]最佳团体

题目梗概

题目给定一棵树，选K个节点。

任意节点满足如果它被选其父亲也必须被选。

使得价值与花费之比最大。

解题思路

这类题目显然是要二分的。

最初的想法，在树上进行左儿子右父亲DP合并，这样每次check的效率是O(n^2)。

但是怎么卡就是不过（我下面给出的就是树上合并的代码）。

我将其归结为常数过大，记住其效率是稳定的。

因此引进DFS序上DP，可以想象常数极小。

$f_{i,j}$表示DFS序的前i个选择j个的最大值。

如果i选，$f_{i,j}-->f_{i+1,j+1}$

如果i不选，则跳过整棵子树$f_{i,j}-->f_{ed[i],j}$

代码实现简单，所以就不给出了，因为博主太弱所以是第一次遇见这种写法。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=2505;
int tot,lnk[maxn],son[2*maxn],nxt[2*maxn];
int s[maxn],p[maxn],K,n,size[maxn];
double v[maxn],f[maxn][maxn],L,R,ans,mid;
void add(int x,int y){nxt[++tot]=lnk[x];lnk[x]=tot;son[tot]=y;}
void merge(int x,int y){
    for (int i=min(K,size[x]);i>=0;i--)
    for (int j=i-1;j>=0&&i-j<=size[y];j--)
    f[x][i]=max(f[x][i],f[x][j]+f[y][i-j]);
}
void DFS(int x,int fa){
    for (int j=lnk[x];j;j=nxt[j]) if (son[j]!=fa){
        DFS(son[j],x);size[x]+=size[son[j]];
        merge(x,son[j]);
    }
    size[x]++;
    if (x) for (int i=min(size[x],K);i>=1;i--) f[x][i]=f[x][i-1]+v[x];
}
bool check(double x){
    for (int i=1;i<=n;i++) v[i]=(double)p[i]-x*s[i];
    memset(f,254,sizeof(f));
    for (int i=0;i<=n;i++) f[i][0]=0;
    DFS(0,0);
    return f[0][K]>=0;
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    scanf("%d%d",&K,&n);
    for (int i=1;i<=n;i++){
        int fa;scanf("%d%d%d",&s[i],&p[i],&fa);
        add(i,fa);add(fa,i);
    }
    L=0;R=1e4;
    while(R-L>=1e-3){
        mid=(R+L)/2;
        if (check(mid)) ans=mid,L=mid;else R=mid;
    }
    printf("%.3lf\n",ans);
    return 0;
}