【C++ 前缀和 状态机dp】2826. 将三个组排序

本文涉及的基础知识点

C++算法：前缀和、前缀乘积、前缀异或的原理、源码及测试用例 包括课程视频
C++动态规划

LeetCode2826. 将三个组排序

给你一个整数数组 nums 。nums 的每个元素是 1，2 或 3。在每次操作中，你可以删除 nums 中的一个元素。返回使 nums 成为 非递减 顺序所需操作数的 最小值。
示例 1：
输入：nums = [2,1,3,2,1]
输出：3
解释：
其中一个最优方案是删除 nums[0]，nums[2] 和 nums[3]。
示例 2：
输入：nums = [1,3,2,1,3,3]
输出：2
解释：
其中一个最优方案是删除 nums[1] 和 nums[2]。
示例 3：
输入：nums = [2,2,2,2,3,3]
输出：0
解释：
nums 已是非递减顺序的。
提示：
1 <= nums.length <= 100
1 <= nums[i] <= 3
进阶：你可以使用 O(n) 时间复杂度以内的算法解决吗？

性质

性质一：如果删除后的数组，包含1，令i1是小标最大的1。则i1之前的2和3都删除，之后的1都删除。
性质二：如果删除后的数组，包括3，令i3是小标最小的3。最i3之前的3都删除，i3之后的1和2删除。0
a[i]=a1+a2。a1 = 1的数量 - a[0…i-1]的数量，即需要删除的1的数量。a2即nums[0…i-1]中2的数量。
b[i]=b3+a2。b3是要删除的3的数量。b2是nums[i+1…]中2的数量。
a2+b2等于删除的2的数量。

前缀和

枚举b[i]，cur = b[i] + min(a[…i-1])
min(cur)就是答案。
注意：nums[i]不为1时，a[i]是非法，可以用个一大数表示比如：1E6。
时间复杂度：O(n)

代码

class Solution {
		public:
			int minimumOperations(vector<int>& nums) {
				vector<int> preSums[3];
				for (int i = 0; i < 3; i++) {
					preSums[i].emplace_back(0);
					for (const auto& n : nums) {
						preSums[i].emplace_back((n == (i+1)) + preSums[i].back());
					}
				}
				vector<int> b(nums.size(), 1000'000);
				for (int i = 0; i < nums.size();i++) {
					b[i] = preSums[2][i] + (preSums[1].back() - preSums[1][i]);
				}
				int a = preSums[0].back(); //1全部删除
				int ret = nums.size()-preSums[1].back();
				for (int i = 0; i < nums.size(); i++) {
					ret = min(ret,a + b[i]);
					a = min(a, (preSums[0].back() - preSums[0][i + 1] + preSums[1][i + 1]));
				}
				ret = min(ret, a + preSums[2].back());
				return ret;
			}
		};

单元测试

	vector<int> nums;
		TEST_METHOD(TestMethod1)
		{
			nums = { 1 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(0, res);
		}
		TEST_METHOD(TestMethod2)
		{
			nums = { 2 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(0, res);
		}
		TEST_METHOD(TestMethod3)
		{
			nums = { 3 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(0, res);
		}
		TEST_METHOD(TestMethod4)
		{
			nums = { 2,2,1,3,2,2 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(2, res);
		}
		TEST_METHOD(TestMethod5)
		{
			nums = { 1,3,2,2 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(1, res);
		}
		TEST_METHOD(TestMethod6)
		{
			nums = { 1,3,2,2 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(1, res);
		}
		TEST_METHOD(TestMethod7)
		{
			nums = { 2,2,1,2,3 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(1, res);
		}
		TEST_METHOD(TestMethod11)
		{
			nums = { 2, 1, 3, 2, 1 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(3, res);
		}	
		TEST_METHOD(TestMethod12)
		{
			nums = { 1,3,2,1,3,3 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(2, res);
		}
		TEST_METHOD(TestMethod13)
		{
			nums = { 2,2,2,2,3,3 };
			auto res = Solution().minimumOperations(nums);
			AssertEx(0, res);
		}

状态机动态规划

动态规划的状态表示

dp1[i][j] 是 num[0…i]以j结尾的最长升序子序列。
pre = dp1[i-1],dp = dp[i]
空间复杂度：O(1)

动态规划的转移方程

dp = pre 不选择nums[i]
MaxSelf(dp[j] , max(pre[0…j])+1) 选择nums[j]

动态规划的初始状态

pre = {0,0,0}

动态规划的填表顺序

两层循环，都是从小到大。

动态规划的返回值

num.size() - max(pre)

代码

class Solution {
		public:
			int minimumOperations(vector<int>& nums) {
				vector<int> pre(3, 0);
				for (auto n : nums) {
					auto dp = pre;
					n--;
					dp[n] = max(dp[n], 1+*max_element(pre.begin(), pre.begin() + n + 1));		
					pre.swap(dp);
				}
				return nums.size() - *max_element(pre.begin(),pre.end());
			}
			int minimumOperations2(vector<int>& nums) {
				vector<int> preSums[3];
				for (int i = 0; i < 3; i++) {
					preSums[i].emplace_back(0);
					for (const auto& n : nums) {
						preSums[i].emplace_back((n == (i + 1)) + preSums[i].back());
					}
				}
				vector<int> b(nums.size(), 1000'000);
				for (int i = 0; i < nums.size(); i++) {
					b[i] = preSums[2][i] + (preSums[1].back() - preSums[1][i]);
				}
				int a = preSums[0].back(); //1全部删除
				int ret = nums.size() - preSums[1].back();
				for (int i = 0; i < nums.size(); i++) {
					ret = min(ret, a + b[i]);
					a = min(a, (preSums[0].back() - preSums[0][i + 1] + preSums[1][i + 1]));
				}
				ret = min(ret, a + preSums[2].back());
				return ret;
			}
		};

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标 及时的反馈 拉伸区（难度合适） 专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功 成功+反思=成功

视频课程

先学简单的课程，请移步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++**实现。