【前言】
非常简单的动态规划题。
【暴力】
搜索,搜出所有可能性。
【不完美的超时做法】
设f[i][j][k]表示,只买前i种商品,共买了j件,能否达到价钱k。
可以转移到f[i+1][j][k]和f[i][j+1][k]
复杂度是O(nm^2*max(v))极限是500^4
【正解】
考虑优化dp。
这里有个小小的技巧。
将所有物品的价格排序,然后减去最小的,这样的作用是,假设我们选的物品个数不足m个,那么可以默认为不足部分全部选择了最小的那一个,我们就可以省去一维,则复杂度变成了O(nm^2*max(v))
参考代码:
class Solution {
public:
/**
*
* @param n int整型
* @param m int整型
* @param b int整型一维数组
* @param bLen int b数组长度
* @return long长整型
*/
int a[250010],f[250010];
long long wwork(int n, int m, int* b, int bLen) {
// write code here
for (int i=1;i<=n;i++) a[i]=b[i-1];
sort(a+1,a+n+1);
for (int i=2;i<=n;i++) a[i]=a[i]-a[1];
int mn=a[1];
a[1]=0;
int mx=a[n]*m;
for (int i=1;i<=250000;i++) f[i]=600;
for (int i=1;i<=n;i++)
{
if (a[i]==a[i-1]) continue;
for (int j=a[i];j<=mx;j++) f[j]=min(f[j],f[j-a[i]]+1);
}
long long ans=0;
for (int i=0;i<=mx;i++)
{
if (f[i]<=m)
{
ans=ans+i+m*mn;
//printf("%d ",i+k*mn);
}
}
return ans;
}
};
京公网安备 11010502036488号