文章目錄

• • 1. 題目
  • 2. 解題

221 / 3117，前7.1%

574 / 9692，前 5.9%

周賽前2題如下：
LeetCode 5641. 卡車上的最大單元數（排序，模擬）
LeetCode 5642. 大餐計數（map計數 + 二分查找）
第4題：LeetCode 5644. 得到子序列的最少操作次數（最長上升子序DP nlogn）

1. 題目

我們稱一個分割整數數組的方案是 好的 ，當它滿足：

• 數組被分成三個 非空 連續子數組，從左至右分別命名為 left ， mid ， right 。
• left 中元素和小于等于 mid 中元素和，mid 中元素和小于等于 right 中元素和。

給你一個 非負 整數數組 nums ，請你返回 好的 分割 nums 方案數目。
由于答案可能會很大，請你將結果對 109 + 7 取余后返回。

示例 1： 輸入：nums = [1,1,1] 輸出：1 解釋：唯一一種好的分割方案是將 nums 分成 [1] [1] [1] 。示例 2： 輸入：nums = [1,2,2,2,5,0] 輸出：3 解釋：nums 總共有 3 種好的分割方案： [1] [2] [2,2,5,0] [1] [2,2] [2,5,0] [1,2] [2,2] [5,0]示例 3： 輸入：nums = [3,2,1] 輸出：0 解釋：沒有好的分割方案。提示： 3 <= nums.length <= 10^5 0 <= nums[i] <= 10^4

https://leetcode-cn.com/problems/ways-to-split-array-into-three-subarrays/

2. 解題

• 二分查找前綴和的切分位置

class Solution { public:int waysToSplit(vector<int>& nums) {int n = nums.size();vector<int> presum(nums);for(int i = 1; i < n; i++)presum[i] += presum[i-1];//前綴和long long ans = 0, mod = 1e9+7;for(int i = 0; i < n-2; i++){int a = presum[i];auto it = lower_bound(presum, i+1, n-2, 2*a);//第二段的前綴和至少為 2a, 在 位置 [i+1, n-2] 內查找 if(it == -1)break;// b = presum[it]-a, c = presum[n-1]-presum[it]// c >= b, presum[it] <= (presum[n-1]+a)/2auto it1 = upper_bound(presum, i+1, n-2, (presum[n-1]+a)/2);if(it1 != -1 && it1>=it)ans = (ans+it1-it+1)%mod;}return ans;}int lower_bound(vector<int>& presum, int L, int R, int t){int l = L, r = R, n = presum.size(), mid;while(l <= r){int mid = (l+r)/2;if(presum[mid] < t)l = mid + 1;else{if(mid==L || (presum[mid-1] < t))return mid;elser = mid - 1;}}return -1;}int upper_bound(vector<int>& presum, int L, int R, int t){int l = L, r = R, n = presum.size(), mid;while(l <= r){int mid = (l+r)/2;if(presum[mid] > t)r = mid - 1;else{if(mid==R || (presum[mid+1] > t))return mid;elsel = mid + 1;}}return -1;} };

368 ms 83.4 MB C++

---

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

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

總結

以上是生活随笔為你收集整理的LeetCode 1712. 将数组分成三个子数组的方案数（前缀和 + 二分查找）的全部內容，希望文章能夠幫你解決所遇到的問題。

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