本文涉及的基础知识点

C++算法：滑动窗口总结

本题其它解法

【C++二分查找 滑动窗口】2831. 找出最长等值子数组|1975

LeetCode2831. 找出最长等值子数组

给你一个下标从 0 开始的整数数组 nums 和一个整数 k 。
如果子数组中所有元素都相等，则认为子数组是一个 等值子数组 。注意，空数组是 等值子数组 。
从 nums 中删除最多 k 个元素后，返回可能的最长等值子数组的长度。
子数组 是数组中一个连续且可能为空的元素序列。
示例 1：
输入：nums = [1,3,2,3,1,3], k = 3
输出：3
解释：最优的方案是删除下标 2 和下标 4 的元素。
删除后，nums 等于 [1, 3, 3, 3] 。
最长等值子数组从 i = 1 开始到 j = 3 结束，长度等于 3 。
可以证明无法创建更长的等值子数组。
示例 2：
输入：nums = [1,1,2,2,1,1], k = 2
输出：4
解释：最优的方案是删除下标 2 和下标 3 的元素。
删除后，nums 等于 [1, 1, 1, 1] 。
数组自身就是等值子数组，长度等于 4 。
可以证明无法创建更长的等值子数组。
提示：
1 <= nums.length <= 10⁵
1 <= nums[i] <= nums.length
0 <= k <= nums.length

C++滑动窗口

n = nums.length
令最优解的第一个下标是i1,最后一个下标是i2。则nums[i1]==nums[i2]，且nums[i1…i2]直接最多k个数$\ne$ nums[i1]。由于是最长子数组，故i2越大越好。i=i1,j=i2+1。
j是以下两种情况之一：
一，j == n。
二，nums[i…j]中不等于nums[i]的数 大于>k且j最大。

代码

核心代码

class Solution {
		public:
			int longestEqualSubarray(vector<int>& nums, int k) {
				const int iMax = *max_element(nums.begin(), nums.end());
				vector<int> cnt(iMax + 1);
				int ans = 0;
				for (int i = 0, j = 0; i < nums.size(); i++) {
					while ((j < nums.size())) {
						if (cnt[nums[i]] + (nums[i] == nums[j]) + k < j + 1 - i) { break; }
						cnt[nums[j]]++; j++;
					}
					ans = max(ans, cnt[nums[i]]);
					cnt[nums[i]]--;
				}
				return ans;
			}
		};

第二版代码

class Solution {
		public:
			int longestEqualSubarray(vector<int>& nums, int k) {
				const int iMax = *max_element(nums.begin(), nums.end());
				vector<int> cnt(iMax + 1);
				int ans = 0;
				for (int i = 0, j = 0; i < nums.size(); i++) {
					while ((j < nums.size())&& (cnt[nums[i]] + (nums[i] == nums[j]) + k >= j + 1 - i)) {	
						cnt[nums[j]]++; j++;
					}
					ans = max(ans, cnt[nums[i]]);
					cnt[nums[i]]--;
				}
				return ans;
			}
		};

单元测试

int  k;
		vector<int> nums;
		TEST_METHOD(TestMethod1)
		{
			nums = { 1,2,3,4 }, k = 3;
			auto res = Solution().longestEqualSubarray(nums, k);
			AssertEx(1, res);
		}
		TEST_METHOD(TestMethod2)
		{
			nums = { 1,1,1,1 }, k = 3;
			auto res = Solution().longestEqualSubarray(nums, k);
			AssertEx(4, res);
		}
		TEST_METHOD(TestMethod11)
		{
			nums = { 1, 3, 2, 3, 1, 3 }, k = 3;
			auto res = Solution().longestEqualSubarray(nums, k);
			AssertEx(3, res);
		}
		TEST_METHOD(TestMethod12)
		{
			nums = { 1,1,2,2,1,1 }, k = 2;
			auto res = Solution().longestEqualSubarray(nums, k);
			AssertEx(4, res);
		}