题目链接:
https://vjudge.net/contest/348156#problem/D
题面:
翻译:
这个问题的标题很熟悉,不是吗?是的,如果你参加过“新秀杯”比赛,你一定看起来是这个冠军。如果你以前没看过,没关系,我会给你一个链接:
链接如下:http://acm.hdu.edu.cn/showproblem.php?pid=2602
今天,我们不希望骨骼的最大值,但第k个最大值的骨头。注意,我们考虑两种方式得到相同的值的骨头是相同的。这意味着,这将是严格的减少序列从第一个最大值,第二个最大值。到第k个最大值。
如果不同值的总数小于K,则输出0。
输入:
第一行包含一个整数T,大小写的数目。
接下来是T个案例,每个案例三行,第一行包含两个整数N,V,K(N<=100,V<=1000,K<=30),代表骨头的数量、包的体积和我们需要的K。第二行包含N个整数,表示每个骨骼的值。第三行包含N个整数,表示每个骨骼的体积。
输出:
每条线表示一个整数的第k个最大值(这个数字将小于2的31次方)。
思路:
作为第一次接触第K优解问题,还是在CSDN里面借鉴了大佬的思路才理解了怎么做:
首先了解一下什么是第K优解;当我们计算正常求最优解时我们只取我们所需要的解,这就导致我们摒弃了许多不优解,而计算第K优解时,就需要把原先摒弃的解按顺序放入一个数组;也就是说在原先求最优解的基础上再加一维数组,若当时是dp[i][j],现在改为dp[i][j][k];若当初为dp[i],现在改为dp[i][k],把所以情况都存储到数组里面去,然后定义一个存储K个,之后每次进行判断,把第K大的始终存储在下标为K的二维数组中,通过3重for循环来实现这个效果,最后就直接输出dp[v}[k]就是第K大的解。
参考代码:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int main()
{
long long t;
scanf("%lld",&t);
while(t--)
{
int n,v,k,i,j,z,a,b,c;
scanf("%d%d%d",&n,&v,&k);
int w[10000];
int val[10000];
int dp[1000][100];
int A[1000];
int B[1000];
memset(dp,0,sizeof(dp));
for(i=0; i<n; i++)
{
scanf("%d",&val[i]);
}
for(i=0; i<n; i++)
{
scanf("%d",&w[i]);
}
for(i=0; i<n; i++)
{
for(j=v; j>=w[i]; j--)
{
for(z=1; z<=k; z++)//把所有情况都存储起来
{
A[z] = dp[j-w[i]][z] + val[i];
B[z] = dp[j][z];
}
A[z] = -1, B[z] = -1;
a = b = c = 1;
while (c<=k && (A[a] != -1 || B[b] != -1))//所有解由最优到不优排列
{
if (A[a] > B[b])
dp[j][c] = A[a++];
else
dp[j][c] = B[b++];
if (dp[j][c] != dp[j][c-1]) //筛选重复解
c++;
}
}
}
printf("%d\n",dp[v][k]);
}
return 0;
}