✨今日算法三题
文章目录
1.左右两边子数组的和相等
题目描述
思路详解
本题我们可以采用遍历解决，记数组的全部元素之和为 total，当遍历到第 i 个元素时，设其左侧元素之和为 sum，则其右侧元素之和为total−numsi−sum。左右侧元素相等即为sum=total−numsi−sum，即 2×sum+numsi=total。
代码与结果
class Solution { public int pivotIndex(int[] nums) { int total = Arrays.stream(nums).sum(); int sum = 0; for (int i = 0; i < nums.length; ++i) { if (2 * sum + nums[i] == total) { return i; } sum += nums[i]; } return -1; } }
2.和可被K整除的子数组
题目描述
思路详解
本题我们可以采用前缀和解决，令P[i]=nums[0]+nums[1]+…+nums[i]。那么每个连续子数组的和sum(i,j) 就可以写成 P[j]−P[i−1]（其中0
代码与结果
class Solution { public int subarraysDivByK(int[] nums, int k) { Map<Integer, Integer> record = new HashMap<Integer, Integer>(); record.put(0, 1); int sum = 0, ans = 0; for (int elem : nums) { sum += elem; // 注意 Java 取模的特殊性，当被除数为负数时取模结果为负数，需要纠正 int modulus = (sum % k + k) % k; int same = record.getOrDefault(modulus, 0); ans += same; record.put(modulus, same + 1); } return ans; } }
3.统计得分小于K的子数组
题目描述
思路详解
本题使用双指针，双指针使用前提：
1.子数组（连续）；
2.有单调性。
本题元素均为正数，这意味着只要某个子数组满足题目要求，在该子数组内的更短的子数组同样也满足题目要求。
做法：枚举子数组右端点，去看对应的合法左端点的个数，那么根据上面的前提 2，我们需要求出合法左端点的最小值。
代码与结果
class Solution { public long countSubarrays(int[] nums, long k) { long ans = 0L, sum = 0L; for (int left = 0, right = 0; right < nums.length; right++) { sum += nums[right]; while (sum * (right - left + 1) >= k) sum -= nums[left++]; ans += right - left + 1; } return ans; } }
✨总结
今天主要练习了前缀和的使用，尤其前缀和的考察一般不会单独出题，往往结合其他算法来达到时间复杂度的优化。
