本文涉及知识点
C++动态规划 状态机dp 性能优化
C++算法:前缀和、前缀乘积、前缀异或的原理、源码及测试用例 包括课程视频
LeetCode 1959. K 次调整数组大小浪费的最小总空间
你正在设计一个动态数组。给你一个下标从 0 开始的整数数组 nums ,其中 nums[i] 是 i 时刻数组中的元素数目。除此以外,你还有一个整数 k ,表示你可以 调整 数组大小的 最多 次数(每次都可以调整成 任意 大小)。
t 时刻数组的大小 sizet 必须大于等于 nums[t] ,因为数组需要有足够的空间容纳所有元素。t 时刻 浪费的空间 为 sizet - nums[t] ,总 浪费空间为满足 0 <= t < nums.length 的每一个时刻 t 浪费的空间 之和 。
在调整数组大小不超过 k 次的前提下,请你返回 最小总浪费空间 。
注意:数组最开始时可以为 任意大小 ,且 不计入 调整大小的操作次数。
示例 1:
输入:nums = [10,20], k = 0
输出:10
解释:size = [20,20].
我们可以让数组初始大小为 20 。
总浪费空间为 (20 - 10) + (20 - 20) = 10 。
示例 2:
输入:nums = [10,20,30], k = 1
输出:10
解释:size = [20,20,30].
我们可以让数组初始大小为 20 ,然后时刻 2 调整大小为 30 。
总浪费空间为 (20 - 10) + (20 - 20) + (30 - 30) = 10 。
示例 3:
输入:nums = [10,20,15,30,20], k = 2
输出:15
解释:size = [10,20,20,30,30].
我们可以让数组初始大小为 10 ,时刻 1 调整大小为 20 ,时刻 3 调整大小为 30 。
总浪费空间为 (10 - 10) + (20 - 20) + (20 - 15) + (30 - 30) + (30 - 20) = 15 。
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 106
0 <= k <= nums.length - 1
动态规划
用K代替k,n = nums.length。
vSpace包括nums所有元素和0,排序并除去重复。
任何时刻,数组的容量一定是vSpace的元素,否则容量-1,仍然符合要求。
动态规划的状态表示
dp’[i][k][j] 表示处理了nums的前i个元素,交换了k次,最终容量为vSpace[j] 浪费的最小容量。
pre = dp’[i]
dp = dp’[i+1]
使用滚动向量的空间复杂度为:O(nK)。
动态规划的转移方程
如果不调整则dp = pre
如果调整,则max(pre[k-1]) ,必须利用前缀和优化。
两种情况都必须:判断dp的j大于等于当前j。
时间复杂度:O(nnk)
动态规划的填表顺序
i,k,j都从小到大。
动态规划的初始值
const int iNotMay = 1e9
除pre[0][0] = 0外,其它全部iNotMay
动态规划的返回值
pre的最小值
注意事项
K++,初始的那次也算。
MinSelf而不是等于。
代码
核心代码
class CDiscretize //离散化
{
public:
CDiscretize(vector<int> nums)
{
sort(nums.begin(), nums.end());
nums.erase(std::unique(nums.begin(), nums.end()), nums.end());
m_nums = nums;
for (int i = 0; i < nums.size(); i++)
{
m_mValueToIndex[nums[i]] = i;
}
}
int operator[](const int value)const
{
auto it = m_mValueToIndex.find(value);
if (m_mValueToIndex.end() == it)
{
return -1;
}
return it->second;
}
int size()const
{
return m_mValueToIndex.size();
}
vector<int> m_nums;
protected:
unordered_map<int, int> m_mValueToIndex;
};
template<class ELE, class ELE2>
void MinSelf(ELE* seft, const ELE2& other)
{
*seft = min(*seft, (ELE)other);
}
class Solution {
public:
int minSpaceWastedKResizing(vector<int>& nums, int K) {
K++;//包括初始的一次
vector<int> vSpace = nums;
vSpace.emplace_back(0);
CDiscretize dis(vSpace);
vector<vector<int>> pre(K+1, vector<int>(dis.size(), m_iNotMay));
pre[0][0] = 0;
for (const auto& n : nums) {
int inx = dis[n];
vector<vector<int>> dp(K + 1, vector<int>(dis.size(), m_iNotMay));
for (int j = inx; j < dis.m_nums.size(); j++) {
for (int k = 0; k <= K; k++) {
MinSelf(&dp[k][j], pre[k][j] + (dis.m_nums[j] - n));
}
}
for (int k = 0; k < K; k++) {
const int iMin = *std::min_element(pre[k].begin(), pre[k].end());
for (int j = inx; j < dis.m_nums.size(); j++) {
MinSelf(&dp[k + 1][j], iMin + (dis.m_nums[j] - n));
}
}
pre.swap(dp);
}
int ret = m_iNotMay;
for (const auto& v : pre) {
ret = min(ret, *min_element(v.begin(), v.end()));
}
return ret;
}
const int m_iNotMay = 1e9;
};
单元测试
template<class T1, class T2>
void AssertEx(const T1& t1, const T2& t2)
{
Assert::AreEqual(t1, t2);
}
template<class T>
void AssertEx(const vector<T>& v1, const vector<T>& v2)
{
Assert::AreEqual(v1.size(), v2.size());
for (int i = 0; i < v1.size(); i++)
{
Assert::AreEqual(v1[i], v2[i]);
}
}
template<class T>
void AssertV2(vector<vector<T>> vv1, vector<vector<T>> vv2)
{
sort(vv1.begin(), vv1.end());
sort(vv2.begin(), vv2.end());
Assert::AreEqual(vv1.size(), vv2.size());
for (int i = 0; i < vv1.size(); i++)
{
AssertEx(vv1[i], vv2[i]);
}
}
namespace UnitTest
{
vector<int> nums;
int k;
TEST_CLASS(UnitTest)
{
public:
TEST_METHOD(TestMethod00)
{
nums = { 10, 20 }, k = 0;
auto res = Solution().minSpaceWastedKResizing(nums, k);
AssertEx(10, res);
}
TEST_METHOD(TestMethod01)
{
nums = { 10,20,30 }, k = 1;
auto res = Solution().minSpaceWastedKResizing(nums, k);
AssertEx(10, res);
}
TEST_METHOD(TestMethod02)
{
nums = { 10,20,15,30,20 }, k = 2;
auto res = Solution().minSpaceWastedKResizing(nums, k);
AssertEx(15, res);
}
TEST_METHOD(TestMethod03)
{
nums = {68873, 132322, 288376, 654645, 663830, 613800, 574994, 746747, 340982, 399408, 626411, 556330, 309223, 475756, 530109, 634837, 451097, 103997, 831622, 735870, 119972, 894218, 686517, 682536, 920342, 806316, 578117, 755856, 907535, 225469, 450962, 473490, 931231, 239593, 433915, 355396, 867612, 475529, 418099, 436197, 765055, 99705, 327731, 447495, 327705, 697417, 267272, 392484, 999915, 160054, 720185, 967820, 497306, 70395, 95657, 872177, 89512, 97759, 452856, 101256, 780759, 774684, 390749, 851202, 136436, 583398, 62895, 614954, 588867, 588968, 345659, 702560, 128169, 430765, 370775, 916044, 697661, 986847, 786731, 443482, 910261, 960332, 756087, 53946, 301602, 629576, 327627, 368728, 108634, 605992, 532061, 223073, 162247, 688638, 505613, 235138, 888433, 113476, 367122, 886439, 583287, 385416, 120466, 799650, 292100, 601988, 237145, 846433, 449957, 886210, 588216, 804712, 858183, 651925, 643379, 707765, 673919, 583784, 27530, 638578, 974216, 689787, 869753, 727158, 280440, 30360, 189180, 286545, 736834, 799445, 323994, 966303, 356332, 368075, 6403, 190218, 912118, 150211, 593831, 689276, 420253
}, k = 37;
auto res = Solution().minSpaceWastedKResizing(nums, k);
AssertEx(22614192, res);
}
};
}
扩展阅读
视频课程
先学简单的课程,请移步CSDN学院,听白银讲师(也就是鄙人)的讲解。
https://edu.csdn.net/course/detail/38771
如何你想快速形成战斗了,为老板分忧,请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176
相关推荐
我想对大家说的话 |
---|
《喜缺全书算法册》以原理、正确性证明、总结为主。 |
按类别查阅鄙人的算法文章,请点击《算法与数据汇总》。 |
有效学习:明确的目标 及时的反馈 拉伸区(难度合适) 专注 |
闻缺陷则喜(喜缺)是一个美好的愿望,早发现问题,早修改问题,给老板节约钱。 |
子墨子言之:事无终始,无务多业。也就是我们常说的专业的人做专业的事。 |
如果程序是一条龙,那算法就是他的是睛 |
测试环境
操作系统:win7 开发环境: VS2019 C++17
或者 操作系统:win10 开发环境: VS2022 C++17
如无特殊说明,本算法用**C++**实现。