LeetCode2588之统计美丽子数组数目(相关话题:前缀异或和)
创始人
2025-05-29 01:04:34
0

题目描述

给你一个下标从 0 开始的整数数组nums 。每次操作中,你可以:

  • 选择两个满足 0 <= i, j < nums.length 的不同下标 i 和 j 。
  • 选择一个非负整数 k ,满足 nums[i] 和 nums[j] 在二进制下的第 k 位(下标编号从 0 开始)是 1 。
  • 将 nums[i] 和 nums[j] 都减去 2k 。

如果一个子数组内执行上述操作若干次后,该子数组可以变成一个全为 0 的数组,那么我们称它是一个 美丽 的子数组。

请你返回数组 nums 中 美丽子数组 的数目。

子数组是一个数组中一段连续 非空 的元素序列。

示例 1:

输入:nums = [4,3,1,2,4]
输出:2
解释:nums 中有 2 个美丽子数组:[4,3,1,2,4] 和 [4,3,1,2,4] 。
- 按照下述步骤,我们可以将子数组 [3,1,2] 中所有元素变成 0 :- 选择 [3, 1, 2] 和 k = 1 。将 2 个数字都减去 21 ,子数组变成 [1, 1, 0] 。- 选择 [1, 1, 0] 和 k = 0 。将 2 个数字都减去 20 ,子数组变成 [0, 0, 0] 。
- 按照下述步骤,我们可以将子数组 [4,3,1,2,4] 中所有元素变成 0 :- 选择 [4, 3, 1, 2, 4] 和 k = 2 。将 2 个数字都减去 22 ,子数组变成 [0, 3, 1, 2, 0] 。- 选择 [0, 3, 1, 2, 0] 和 k = 0 。将 2 个数字都减去 20 ,子数组变成 [0, 2, 0, 2, 0] 。- 选择 [0, 2, 0, 2, 0] 和 k = 1 。将 2 个数字都减去 21 ,子数组变成 [0, 0, 0, 0, 0] 。

示例 2:

输入:nums = [1,10,4]
输出:0
解释:nums 中没有任何美丽子数组。

提示:

  • 1 <= nums.length <= 105
  • 0 <= nums[i] <= 106

解题思路

  1. 相同的数相互异或等于0
  2. 0和任何数异或都等于它本身

由于每次可以执行的操作是选择数组中某一位均为1的两个数,将其减去2^{k} ,于是可以想到位运算的异或操作,相同的数异或结果为0


如果子数组异或结果为0,那么该子数组为美丽子数组,因此可以将题意转换为,求出子数组异或结果为0的个数

借鉴前缀和数组的概念,我们可以记录nums[i]对应的异或数组,假设nums[0,i]对应的异或值为a,nums[0,j]对应的异或值也为为a,那么nums[i+1,j]对应的异或值为0【任何一个数与0异或的结果还为0】

因此,我们枚举子数组的每一个右端点iii,将nums[0,i-1]的异或结果存放在哈希表中,哈希表的键值为异或结果,key值为出现的次数,那么以nums[i]为右端点的美丽子数组的数目即为其对应的异或结果在这之前出现的次数

代码实现 

class Solution {public long beautifulSubarrays(int[] nums) {long res = 0L;Map map = new HashMap<>();map.put(0, 1L);int cur = 0;for (int i = 0; i < nums.length; i++){cur ^= nums[i];res += map.getOrDefault(cur, 0L);map.put(cur, map.getOrDefault(cur, 0L) + 1);}return res;}
}

题目拓展

给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:

  • 子数组大小 至少为 2 ,且
  • 子数组元素总和为 k 的倍数。

如果存在,返回 true ;否则,返回 false 。

如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。0 始终视为 k 的一个倍数。

解题思路

同余定理:如果两个整数m、n满足n-m能被k整除,那么n和m对k同余

即 ( pre(j) - pre (i) ) % k == 0 则 pre(j) % k == pre(i) % k

  • 计算前缀和 pre( j ) % k

  • 当pre(j) % k 在哈希表中已存在,则说明此时存在 i 满足 pre(j) % k == pre(i) % k ( i < j )HashMap里,已知Key,可以取到Value 即i的值, 最后 判断 j - i >= 2 是否成立 即可

  • 当 pre(j) % k 不存在于哈希表,则将 (pre(j) % k, j ) 存入哈希表

因在计算 pre(i) = (pre(i-1) + nums[i]) % k 时,pre(i) 只与上一个状态有关

故可以直接用变量pre 替代数组。 那么求前缀和 % k 的公式就简化为题解代码中的 remainder = (remainder + nums[i]) % k;

class Solution {public boolean checkSubarraySum(int[] nums, int k) {int m = nums.length;if (m < 2) {return false;}Map map = new HashMap();map.put(0, -1);int remainder = 0;for (int i = 0; i < m; i++) {remainder = (remainder + nums[i]) % k;if (map.containsKey(remainder)) {int prevIndex = map.get(remainder);if (i - prevIndex >= 2) {return true;}} else {map.put(remainder, i);}}return false;}
}

相关题目

LeetCode560之和为K的子数组(相关话题:前缀和)_击水三千里的博客-CSDN博客
1590. 使数组和能被 P 整除
525. 连续数组
面试题 17.05. 字母与数字
1915. 最美子字符串的数目
1371. 每个元音包含偶数次的最长子字符串
1542. 找出最长的超赞子字符串

 

相关内容

热门资讯

监控摄像头接入GB28181平... 流程简介将监控摄像头的视频在网站和APP中直播,要解决的几个问题是:1&...
Windows10添加群晖磁盘... 在使用群晖NAS时,我们需要通过本地映射的方式把NAS映射成本地的一块磁盘使用。 通过...
protocol buffer... 目录 目录 什么是protocol buffer 1.protobuf 1.1安装  1.2使用...
Fluent中创建监测点 1 概述某些仿真问题,需要创建监测点,用于获取空间定点的数据࿰...
educoder数据结构与算法...                                                   ...
MySQL下载和安装(Wind... 前言:刚换了一台电脑,里面所有东西都需要重新配置,习惯了所...
MFC文件操作  MFC提供了一个文件操作的基类CFile,这个类提供了一个没有缓存的二进制格式的磁盘...
在Word、WPS中插入AxM... 引言 我最近需要写一些文章,在排版时发现AxMath插入的公式竟然会导致行间距异常&#...
有效的括号 一、题目 给定一个只包括 '(',')','{','}'...
【Ctfer训练计划】——(三... 作者名:Demo不是emo  主页面链接:主页传送门 创作初心ÿ...