01背包 第k優解


求次優解、第K 優解

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

struct Node
{
    int price;
    int val;
} node[1005];

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,v,k,i,dp[1005][31] = {0},a[31],b[31];
        scanf("%d%d%d",&n,&v,&k);
        for(i = 0; i<n; i++)
            scanf("%d",&node[i].price);
        for(i = 0; i<n; i++)
            scanf("%d",&node[i].val);
        int j;
        for(i = 0; i<n; i++)
        {
            for(j = v; j>=node[i].val; j--)
            {
                int cnt = 0,d;
                for(d = 1; d<=k; d++)//分別將放入第i個石頭與不放第i個石頭的結果存入a,b,數組之中
                {
                    a[d] = dp[j-node[i].val][d]+node[i].price;
                    b[d] = dp[j][d];
                }
                int x,y,z;
                x = y = z = 1;
                a[d] = b[d] = -1;
                while(z<=k && (x<=k || y<=k))//循環找出前K個的最優解
                {
                    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[v][k]);
    }

    return 0;
}



注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



 
粤ICP备14056181号  © 2014-2021 ITdaan.com