摇钱树 dp

原题：zjnu 1209

题意：

有n堆钱，给出每堆的初始数量和消失速度，你可以拿走k堆钱，每拿走一堆，所有剩下的钱就会按照消失速度减少，求k天所能拿的最大金额

解析：

拿哪堆最好呢？第一个因素是初始金额大，第二个因素是消失速度小，而到底怎么在这两种因素中做一个抉择，是这题的难点

假设n堆我们都可以拿走，那么肯定是先 拿消失得快的那堆，
那如果现在多出第n+1堆，且比前n堆的消失速度都小呢？
最优结果是不是有两种：

1. 第n+1堆太少了不需要 —> 相当于n堆拿n堆
2. 需要 ，前n堆拿n-1堆再加上第n+1堆，并且因为n+1消失速度最小，n+1堆最后拿
   即：n堆拿n-1 + sum[n+1]-(n-1)*v[n+1]

所以我们按照这个思路，把所有堆按消失速度降序排列，用dp[i][j]表示前i堆拿j堆的最优解，状态转移方程：dp[i][j] = max( dp[i-1][j] , dp[i-1][j-1] + sum[i]-(j-1)*v[i] )

最后要注意的点是，因为dp[i][j]和dp[i][j-1]并不像其他dp一样有直接关系，而是只从dp[i-1]推过来，所以最后答案应该是dp[n][1 to k]中取最大值

代码：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#define N 1009
#include<string.h>
using namespace std;

int n,k;
struct node{
    int sum,lo;
    bool operator < (const node k)const{
        return lo>k.lo;
    }
}e[N];

int dp[N][N];
int main(){
    while(scanf("%d%d",&n,&k),n||k){
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)scanf("%d",&e[i].sum);
        for(int i=1;i<=n;i++)scanf("%d",&e[i].lo);
        sort(e+1,e+1+n);
        for(int i=1;i<=n;i++){
            for(int j=1;j<=k;j++){
                if(j>i)break;
                int add=e[i].sum-(j-1)*e[i].lo;
                if(add<0)add=0;
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-1]+add);
                //printf("dp[%d][%d] is %d\n",i,j,dp[i][j]);
            }
        }
        int ans=0;
        //because dp[i][j] just change from dp[i-1][?],so you can't judge dp[n][?] is biggest
        for(int i=1;i<=k;i++)ans=max(ans,dp[n][i]);
        printf("%d\n",ans);
    }
}