题目梗概
题目给定一棵树,选K个节点。
任意节点满足如果它被选其父亲也必须被选。
使得价值与花费之比最大。
解题思路
这类题目显然是要二分的。
最初的想法,在树上进行左儿子右父亲DP合并,这样每次check的效率是O(n^2)。
但是怎么卡就是不过(我下面给出的就是树上合并的代码)。
我将其归结为常数过大,记住其效率是稳定的。
因此引进DFS序上DP,可以想象常数极小。
fi,j 表示DFS序的前i个选择j个的最大值。
如果i选, fi,j−−>fi+1,j+1
如果i不选,则跳过整棵子树 fi,j−−>fed[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;
}