力扣每日一题:使数组和能被p整除

给你一个正整数数组 nums,请你移除 最短 子数组(可以为 ),使得剩余元素的 能被 p 整除。 不允许 将整个数组都移除。

请你返回你需要移除的最短子数组的长度,如果无法满足题目要求,返回 -1

子数组 定义为原数组中连续的一组元素。

思路:

首先,这个题是要去掉某个子数组,因此要考虑用到前缀和的思想。

其次,这里是尝试去掉一部分后,剩下的部分余p等于0,也就是说,剩下的部分,和去掉的部分,余数是相同的。

那么,我们假设数组的和余p为x,要去除的区间是l,r,则有x=(sumr-suml)%p,而我们在遍历数组时,可以通过一次O(N)的时间复杂度得到x,后续遍历过程中,O(1)的时间复杂度得到sumr,那么,如果我们可以快速定位suml,就可以很快的找到符合条件的子数组。那么,如何快速找到这个suml,我们可以发现,这里要的是最短子数组,那么,suml一定是所有能符合x=(sumr-suml)%p中,最大的那个l。因此我们可以每次记录当前的前缀和余p,去更新对应位置。然后,每次只需要快速查找是否有符合条件的那个l,有就更新最小数组长度。

然后,我们来说一下这里的suml,suml就是((sumr-x)%p+p)%p,这里是确保一定出来的是一个正数。

要注意的是,这里如果用数组记录前缀和余p,会出现内存爆了的情况。

复制代码
class Solution {
public:
    int minSubarray(vector<int>& nums, int p) {
        int n = nums.size();
        vector<int> sum(n + 1, 0);
        for (int i = 0; i < n; i++)
            sum[i + 1] = (sum[i] + nums[i])%p;

        int x = sum[n] % p; // 表示两个区间模k余mm  (b-a)mod p=mm   bmodp -amodp==mm
        if (x == 0)
            return 0;

        for (int i = 0; i < n; i++) {
            if (nums[i] % p == x)
                return 1;
        }

        unordered_map<int,int> mxlen;// 记录当前模p余i的最大坐标
        int ans = n;
        for (int i = 0; i <= n; i++) {
           mxlen[sum[i]]=i;
           auto it=mxlen.find( ((sum[i]-x)%p+p)%p  );
           if(it!=mxlen.end())
           ans=min(ans,i-it->second);
        }
        if (ans==n)
            return -1;
        return ans;
    }
};
相关推荐
祖力558 小时前
数据结构的基本概念与单向链表(链表数据类型构造、创建、头插、尾插、头删、尾删)
数据结构·链表
大明者省10 小时前
WSL2 Ubuntu22.04 GPU训练环境配置指南
人工智能·算法·计算机视觉
白狐_79811 小时前
408数据结构第8章:排序②——性质对比秒杀、场景选择与外部排序
java·数据结构·算法
zander25812 小时前
LeetCode 84:柱状图中的最大矩形——单调栈如何确定左右边界
java·数据结构·算法
Shell运维手记12 小时前
Linux 常用基础命令学习笔记
linux·运维·笔记·学习·算法·github
杨航 AI12 小时前
O(n log n):线性对数原理拆解 这个是排序算法的黄金复杂度之一。
算法·排序算法
船厂电气自动化ai大模型13 小时前
AI大模型与数学 第32课 函数凹凸性与二阶导数:拐点求解、凹凸区间计算(10道二阶导数计算题)
数据结构·人工智能·python·深度学习·算法
旖旎夜光13 小时前
LeetCode 3:无重复字符的最长子串(滑动窗口) —— 题解
数据结构·c++·算法·leetcode·滑动窗口
叠层归一研究院14 小时前
AGI 系统(十一):二阶网络 — 二阶递归 × 多主体 (元符号网络)
开发语言·人工智能·算法·php·agi
zlinear数据采集卡15 小时前
D223的PWM电机控制:6路独立脉冲+加减速算法深度解析
arm开发·stm32·嵌入式硬件·算法·fpga开发·架构