CodeForces - 366C Dima and Salad（dp）

C. Dima and Salad

题意：给n个水果和k值，每个水果有a和b两种属性。选一些水果使suma比sumb为k。求max suma；

题解：求出每个水果的x = a-b*k。当选中的水果的x值之和为0时，就是比值为k的条件，因此题目就转换为01满背包问题，dp【0】就是答案。a就是每个水果的价值。

 

方法1：分别dp x值为正数和负数时，dp[j+x[i]] = max(dp[j]+a[i],dp[j+x[i]]);

#include <bits/stdc++.h>
using namespace std;
int a[105], b[105], x[105];
int dp[20000],pp[20000];
int main()
{
//    freopen("in.txt", "r", stdin);
    memset(dp, -0x3f, sizeof(dp));
    memset(pp, -0x3f, sizeof(pp));
    int n, k;
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
    for(int i = 1; i <= n; i++)
    {
        scanf("%d", &b[i]);
        x[i] = a[i] - k * b[i];
    }
    dp[0]=pp[0]=0;
    for(int i = 1; i<=n; i++)
    {
        if(x[i]>=0)
        {
            for(int j = 10000-x[i]; j>=0; j--)
            {
                dp[j+x[i]] = max(dp[j]+a[i],dp[j+x[i]]);
            }
        }
        else
        {
            int t = -x[i];
            for(int j = 10000-t; j>=0; j--)
            {
                pp[j+t] = max(pp[j]+a[i],pp[j+t]);
            }
        }
    }
    int ans = -1;
    for(int i = 0; i<=10000; i++)
    {
        if(dp[i]!=0||pp[i]!=0)
            ans = max(ans,dp[i]+pp[i]);
    }
    printf("%d\n",ans);
    return 0;
}

方法2：如果直接dp的话要dp【i】的i可能i负数，我用map模拟时，不是超时就是wa，之后想到可以将i每个加maxn，但是要注意你的dp数组为一维时，不注意的话会出现一个水果选2次以上的情况（因为x【i】可能为正的或者负的所以要分开讨论）。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e3+5;
const int maxn1 = 1e5;
const int maxn2 = 2e5;
int dp[5000005];
int a[maxn], b[maxn], x[maxn];
int main()
{
//    freopen("in.txt", "r", stdin);
    int n, k;
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
    for(int i = 1; i <= n; i++)
    {
        scanf("%d", &b[i]);
        x[i] = a[i] - k * b[i];
    }
    memset(dp, -0x3f, sizeof(dp));
    dp[maxn1] = 0;
    int t;
    for(int i = 1; i <= n; i++)
    {
        if(x[i] < 0)
        {
            for(int j = 0; j <= maxn2; j++)
                if(j + x[i] >= 0 && j + x[i] <= maxn2)
                    dp[j + x[i]] = max(dp[j + x[i]], dp[j] + a[i]);
        }
        else
        {
            for(int j = maxn2; j >= 0; j--)
                if(j + x[i] >= 0 && j + x[i] <= maxn2)
                    dp[j + x[i]] = max(dp[j + x[i]], dp[j] + a[i]);
        }
    }
    if(dp[maxn1] > 0)
        printf("%d\n", dp[maxn1]);
    else printf("%d\n", -1);
    return 0;
}