SDUTOJ 3307 a(k大背包)

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<math.h>
#include<queue>
#define ll long long
#define inf 1000010+10
using namespace std;
int p[1010],c[1010],a[1010],b[1010];
int dp[1010][1010];
int main()
{
    int n,m,i,j,k;
    int cla;
    scanf("%d",&cla);
    while(cla--)
    {
        scanf("%d%d",&n,&m);
        memset(dp,0,sizeof(dp));
        for(i=1;i<=m;i++)
        {
            scanf("%d %d",&p[i],&c[i]);
        }
        for(i=1;i<=m;i++)
        {
            for(j=n;j>=p[i];j--)
            {
                for(int t=1;t<=2;t++)
                {
                    a[t]=dp[j][t];
                    b[t]=dp[j-p[i] ][t]+c[i];
                }
                int x=1,y=1,z=1;
                a[3]=b[3]=-1;
                while(z<=2&&(x<=2||y<=2))
                {
                    if(a[x]>b[y])
                    {
                        dp[j][z]=a[x],x++;
                    }
                    else
                        dp[j][z]=b[y],y++;
                    if(dp[j][z]!=dp[j][z-1])
                        z++;
                }
            }
        }
        printf("%d\n",dp[n][1]+dp[n][2]);
    }
    return 0;
}

你可能感兴趣的:(01背包第K大背包)