Schrödinger's Knapsack ZOJ - 4019 （DP）

Schrödinger's Knapsack

题目连接：ZOJ - 4019

题意：有两类物品， 价值分别为k1， k2，第一类物品有n个， 第二类有m个， 将其放入容量为c的容器中， 每放一个物品得到价值为：val=k*v（k为放入的物品的价值， v为放入该物品后容器剩余容量）；求得到的价值最大；

思路：将物品分类排序， 那么放入容器的物品一定是每类中最小的优先， 即：若一类中， 第二小的可以放入， 那么最小的必定能放入容器；

#include <iostream>
#include <map>
#include <string.h>
#include <stdio.h>
#include <algorithm>
using namespace std;
const int maxn=2005;
long long ans, dp[maxn][maxn];
long long a[maxn], b[maxn], c, k1, k2;
long long sum1[maxn], sum2[maxn];
int n, m;
int main(){
    int T;
    scanf("%d", &T);
    while(T--){
        ans=0;
        scanf("%lld%lld%lld", &k1, &k2, &c);
        scanf("%d%d", &n, &m);
        for(int i=1; i<=n; i++){
            scanf("%lld", &a[i]);
        }
        for(int j=1; j<=m; j++){
            scanf("%lld", &b[j]);
        }
        sort(a+1, a+n+1);
        sort(b+1, b+1+m);
        sum1[0]=sum2[0]=0;
        for(int i=1; i<=n; i++){
            sum1[i]=sum1[i-1]+a[i];
        }
        for(int j=1; j<=m; j++){
            sum2[j]=sum2[j-1]+b[j];
        }
        //dp数组初始化， 不用memset， 因为每次初始化不需要将所有值都初始化， 只初始化用到的值即可；
        for(int i=0; i<=n; i++){
            for(int j=0; j<=m; j++)
                dp[i][j]=0;
        }
        for(int i=0; i<=n; i++){
            for(int j=0; j<=m; j++){
                if(c-sum1[i]-sum2[j]<0) break;
                if(i==0&&j==0) continue;
                if(i>0) dp[i][j]=max(dp[i][j], dp[i-1][j]+k1*(c-sum1[i]-sum2[j]));
                if(j>0) dp[i][j]=max(dp[i][j], dp[i][j-1]+k2*(c-sum1[i]-sum2[j]));
                ans=max(ans, dp[i][j]);
            }
        }
        printf("%lld\n", ans);
    }
    return 0;
}