README.md

2395. 和相等的子数组

English Version

题目描述

给你一个下标从 0 开始的整数数组 nums ，判断是否存在 两个 长度为 2 的子数组且它们的 和 相等。注意，这两个子数组起始位置的下标必须 不相同 。

如果这样的子数组存在，请返回 true，否则返回 false 。

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

 

示例 1：

输入：nums = [4,2,4]
输出：true
解释：元素为 [4,2] 和 [2,4] 的子数组有相同的和 6 。

示例 2：

输入：nums = [1,2,3,4,5]
输出：false
解释：没有长度为 2 的两个子数组和相等。

示例 3：

输入：nums = [0,0,0]
输出：true
解释：子数组 [nums[0],nums[1]] 和 [nums[1],nums[2]] 的和相等，都为 0 。
注意即使子数组的元素相同，这两个子数组也视为不相同的子数组，因为它们在原数组中的起始位置不同。

 

提示：

• 2 <= nums.length <= 1000
• -109 <= nums[i] <= 109

解法

方法一：哈希表

用哈希表 s 记录数组相邻两元素的和。

遍历数组 nums，若 s 中存在 nums[i] + nums[i + 1]，则返回 true；否则将 nums[i] + nums[i + 1] 加入 s 中。

时间复杂度 $O(n)$，空间复杂度 $O(n)$，其中 $n$ 为数组 nums 的长度。

Python3

class Solution:
    def findSubarrays(self, nums: List[int]) -> bool:
        s = set()
        for a, b in pairwise(nums):
            if (v := a + b) in s:
                return True
            s.add(v)
        return False

Java

class Solution {
    public boolean findSubarrays(int[] nums) {
        Set<Integer> s = new HashSet<>();
        for (int i = 0; i < nums.length - 1; ++i) {
            int v = nums[i] + nums[i + 1];
            if (!s.add(v)) {
                return true;
            }
        }
        return false;
    }
}

C++

class Solution {
public:
    bool findSubarrays(vector<int>& nums) {
        unordered_set<int> s;
        for (int i = 0; i < nums.size() - 1; ++i) {
            int v = nums[i] + nums[i + 1];
            if (s.count(v)) return true;
            s.insert(v);
        }
        return false;
    }
};

Go

func findSubarrays(nums []int) bool {
	s := map[int]bool{}
	for i := 0; i < len(nums)-1; i++ {
		v := nums[i] + nums[i+1]
		if s[v] {
			return true
		}
		s[v] = true
	}
	return false
}

TypeScript

...