文章目錄

• • 1. 題目
  • 2. 解題

1. 題目

給定一個有 n 個整數的數組，你需要找到滿足以下條件的三元組 (i, j, k) ：

• 0 < i, i + 1 < j, j + 1 < k < n - 1
• 子數組 (0, i - 1)，(i + 1, j - 1)，(j + 1, k - 1)，(k + 1, n - 1) 的和應該相等。

這里我們定義子數組 (L, R) 表示原數組從索引為L的元素開始至索引為R的元素。

示例: 輸入: [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注意: 1 <= n <= 2000。 給定數組中的元素會在 [-1,000,000, 1,000,000] 范圍內。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/split-array-with-equal-sum
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

class Solution { public:bool splitArray(vector<int>& nums) {int n = nums.size(), i, j, k;if(n < 7) return false;vector<int> sum(nums);for(i = 1; i < n; i++)sum[i] += sum[i-1];for(j = 3; j < n-3; j++){unordered_set<int> s;for(i = 1; i < j-1; i++){if(sum[i-1] == sum[j-1]- sum[i])s.insert(sum[i-1]);}for(k = j+2; k < n-1; k++){if(sum[k-1]-sum[j] == sum[n-1]-sum[k]&& s.count(sum[k-1]-sum[j]))return true;}}return false;} };

572 ms 11 MB

參考大佬的題解剪枝

$sum(0,i?1)+sum(i+1,j?1)=sum(j+1,k?1)+sum(k+1,n?1)$
$=>sum(0,j?1)?nums[i]=sum(j+1,n?1)?nums[k]$
$=>sum(0,j?1)?sum(j+1,n?1)=nums[i]?nums[k]$
$=>|sum(0,j?1)?sum(j+1,n?1)|=|nums[i]?nums[k]|<=\max ?\min$

作者：palading
鏈接：https://leetcode-cn.com/problems/split-array-with-equal-sum/solution/c-qian-zhui-he-jian-zhi-by-palading/
來源：力扣（LeetCode） 著作權歸作者所有。商業轉載請聯系作者獲得授權，非商業轉載請注明出處。

class Solution { public:bool splitArray(vector<int>& nums) {int n = nums.size(), i, j, k;if(n < 7) return false;vector<int> sum(nums);for(i = 1; i < n; i++)sum[i] += sum[i-1];int maxnum = *max_element(nums.begin(), nums.end());int minnum = *min_element(nums.begin(), nums.end());for(j = 3; j < n-3; j++){if(abs(sum[n-1]-sum[j]-sum[j-1]) > maxnum-minnum)continue;unordered_set<int> s;for(i = 1; i < j-1; i++){if(sum[i-1] == sum[j-1]- sum[i])s.insert(sum[i-1]);}for(k = j+2; k < n-1; k++){if(sum[k-1]-sum[j] == sum[n-1]-sum[k]&& s.count(sum[k-1]-sum[j]))return true;}}return false;} };

8 ms 10.4 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的LeetCode 548. 将数组分割成和相等的子数组（哈希set）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。