P1642 规划

链接
01分数规划+树形DP

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<functional>
#include<iomanip>
using namespace std;
#define in(x) scanf("%d",&x)
#define onword(n,arr) \
for(int i=0;i<n;i++) \
in(arr[i]);
#define foo EOF
#define hear(x) printf("%f ",x);
#define escape ' '
double arr3[101];
void specialin(int *arr1,int *arr2,double k,int n){
for(int i=0;i<n;i++)
arr3[i]=arr1[i]*1.0-k*arr2[i];
}
typedef double fly;
#define equal 1e-7
#define when while
void  show(double *arr,int n)  {
for(int i=0;i<n;i++)
hear(arr[i]);
printf("\n");
}
bool cmp(double a,double b)
{
    return a>b;

}
void quicksortd( double* be, double * ed)
{


    if(be>=ed)return ;
    double *l=be,*r=ed;
    while(l<r)
    {
        while(l<r&&*l>*r)l++;
        if(l<r)
        {
            swap(*l,*r);
            r--;
        }
        while(l<r&&*l>*r)r--;
        if(l<r)
        {
            swap(*l,*r);
            l++;
        }
    }
    quicksortd(be,l-1);
    quicksortd(l+1,ed);
}
bool judge(double *arr,int n,int y)
{
    quicksortd(arr,arr+n-1);
    double sum=0;
    for(int i=0;i<y;i++)
        sum+=arr[i];
    return sum>0;
}
int adj[101],nxt[202],go[202],num=0;
inline void add(int x,int y)
{
    nxt[++num]=adj[x];
    adj[x]=num;
    go[num]=y;
}
inline void adds(int x,int y)
{
    add(x,y);
    add(y,x);
}
double dp[101][101];int nVe[101];
int x,y;
int vis[101];
void dfs(int father,int son)
{
    nVe[son]=1;
    dp[son][0]=0;
    for(int i=adj[son];i;i=nxt[i])
    {
        int sons=go[i];
        if(sons==father)continue;
        dfs(son,sons);
        nVe[son]+=nVe[sons];
        for(int j=min(y,nVe[son]);j>=0;j--)
            for(int k=0;k<=min(j,nVe[sons]);k++)
                dp[son][j]=max(dp[son][j],dp[son][j-k]+dp[sons][k]);
     }
    for(int j=min(y,nVe[son]);j;j--)
    dp[son][j]=dp[son][j-1]+arr3[son];
}
int arr1[101],arr2[101];
inline bool judge(double k,int n)
{
    for(int i=0;i<n;i++)
    arr3[i+1]=arr1[i]*1.0-arr2[i]*k;
    dfs(0,1);
    for(int i=1;i<=x;i++)
    if(dp[i][y]>0){return true;}
    return false;
}
int main()
{
    in(x);
    in(y);
    y=x-y;
    onword(x,arr1);
    onword(x,arr2);
    int a,b;
    for(int i=1;i<x;i++)
    {
        in(a);
        in(b);
        adds(a,b);
    }
    fly l=0,r=1e5;
    when(r-l>=equal)
    {memset(dp,-0x3f3f3f3f,sizeof(dp));
    memset(vis,0,sizeof(vis));
    memset(nVe,0,sizeof(nVe));
        fly mid=(l+r)/2;
        if(judge(mid,x))
            l=mid;
        else r=mid;
     
    
    }
    cout fixed<<setprecision(1) <<r<< endl;
}