0548：将数组分割成和相等的子数组（★★）

力扣第 548 题

题目

给定一个有 n 个整数的数组 nums ，如果能找到满足以下条件的三元组 (i, j, k) 则返回 true ：

1. 0 < i, i + 1 < j, j + 1 < k < n - 1
2. 子数组 (0, i - 1) ， (i + 1, j - 1) ， (j + 1, k - 1) ， (k + 1, n - 1) 的和应该相等。

这里我们定义子数组 (l, r) 表示原数组从索引为 l 的元素开始至索引为 r 的元素。

示例 1:

输入: nums = [1,2,1,2,1,2,1]
输出: True
解释:
i = 1, j = 3, k = 5.
sum(0, i - 1) = sum(0, 0) = 1
sum(i + 1, j - 1) = sum(2, 2) = 1
sum(j + 1, k - 1) = sum(4, 4) = 1
sum(k + 1, n - 1) = sum(6, 6) = 1

示例 2:

输入: nums = [1,2,1,2,1,2,1,2]
输出: false

提示:

• n == nums.length
• 1 <= n <= 2000
• -106 <= nums[i] <= 106

分析

• 类似 0454，预处理两块的哈希表，再遍历剩下的两块

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10

class Solution:
    def splitArray(self, nums: List[int]) -> bool:
        n = len(nums)
        P = list(accumulate(nums))
        for j in range(3,n-3):
            vis = {P[i-1] for i in range(1,j-1) if P[i-1]==P[j-1]-P[i]}
            for k in range(j+2,n-1):
                if P[k-1]-P[j]==P[n-1]-P[k] and P[k-1]-P[j] in vis:
                    return True
        return False

979 ms