LeetCode 2831.找出最长等值子数组(cpp, python3)

题目链接:2831.找出最长等值子数组

> 将元素值为下标索引,将元素在数组nums中的出现位置记录在二维数组position中,position每个下标对应的数组都是下标索引在nums中的出现位置

> 记position的下标为index,index在nums中的出现次数为position[index].size(),记作cnt1,而在nums数组中index第一次出现到最后一次出现,这个范围内nums的元素个数为:position[index][position[index].size() - 1] - position[index][0] + 1,记作cnt2,那么cnt2 - cnt1就等于需要删除的元素的个数,删除之后最长等值数组的长度就等于滑动窗口的right - left + 1

cpp实现:

cpp 复制代码
class Solution {
public:
    int longestEqualSubarray(vector<int>& nums, int k) {
        int ans = 0; // 初始化长度为0
        vector<vector<int>> position(nums.size() + 1); //定义nums中元素出现的下标数组
        for (int i = 0; i < nums.size(); i++) {
            position[nums[i]].push_back(i); 
        }
        for (auto& e : position) { //分别遍历每一个元素所对应的下标
            int left = 0; //滑动窗口初始化左端点为0
            for (int right = 0; right < e.size(); right++) { //不断移动右端点
                while (e[right] - e[left] + 1 - (right - left + 1) > k) { //当需要移除的元素个数大于k时
                    left += 1; //收缩窗口
                }
                ans = max(ans, right - left + 1); //退出循环之后,说明满足条件,更新ans
            }
        }
        return ans;
    }
};

python3实现:

python 复制代码
class Solution:
    def longestEqualSubarray(self, nums: List[int], k: int) -> int:
        ans = 0
        position = [[] for _ in range(len(nums) + 1)]
        for i, x in enumerate(nums):
            position[x].append(i)
        for e in position:
            left = 0
            for right in range(len(e)):
                while e[right] - e[left] + 1 - (right - left + 1) > k:
                    left += 1
                ans = max(ans, right - left + 1)
        return ans
相关推荐
豌豆花下猫3 分钟前
Python 潮流周刊#110:JIT 编译器两年回顾,AI 智能体工具大爆发(摘要)
后端·python·ai
我喜欢就喜欢13 分钟前
RapidFuzz-CPP:高效字符串相似度计算的C++利器
开发语言·c++
千帐灯无此声14 分钟前
Linux 测开:日志分析 + 定位 Bug
linux·c语言·c++·bug
莫彩15 分钟前
【Modern C++ Part7】_创建对象时使用()和{}的区别
开发语言·c++
课堂剪切板44 分钟前
ch07 题解
算法·深度优先
June bug1 小时前
【Python基础】变量、运算与内存管理全解析
开发语言·python·职场和发展·测试
蹦蹦跳跳真可爱5891 小时前
Python----OpenCV(几何变换--图像平移、图像旋转、放射变换、图像缩放、透视变换)
开发语言·人工智能·python·opencv·计算机视觉
蹦蹦跳跳真可爱5891 小时前
Python----循环神经网络(Transformer ----Layer-Normalization(层归一化))
人工智能·python·rnn·transformer
m0_625686551 小时前
Day58
python
夜阳朔1 小时前
Conda环境激活失效问题
人工智能·后端·python