开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第24天,点击查看活动详情
一、题目描述:
1013. 将数组分成和相等的三个部分 - 力扣(LeetCode)
给你一个整数数组 arr,只有可以将其划分为三个和相等的 非空 部分时才返回 true,否则返回 false。
形式上,如果可以找出索引 i + 1 < j 且满足 (arr[0] + arr[1] + ... + arr[i] == arr[i + 1] + arr[i + 2] + ... + arr[j - 1] == arr[j] + arr[j + 1] + ... + arr[arr.length - 1]) 就可以将数组三等分。
示例 1:
输入:arr = [0,2,1,-6,6,-7,9,1,2,0,1]
输出:true
解释:0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1
示例 2:
输入:arr = [0,2,1,-6,6,7,9,-1,2,0,1]
输出:false
示例 3:
输入:arr = [3,3,6,5,-2,2,5,1,-9,4]
输出:true
解释:3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4
提示:
- 3 <= arr.length <= 5 * 10^4
- -104 <= arr[i] <= 10^4
二、思路分析:
先对数组求和,如果所求之和不是 33 的倍数,直接返回false 双指针left和right分别从数组两端开始向中间累加,直到部分和等于总和的 1 / 31/3 或越过边界 最后通过边界条件和两边的部分和是否相等判断返回结果
三、AC 代码:
class Solution {
public:
bool canThreePartsEqualSum(vector<int>& A) {
int sum = accumulate(A.begin(), A.end(), 0);
if (sum % 3) return false;
int left = 0, right = A.size() - 1, sl = 0, sr = 0, t = sum / 3;
do sl += A[left++]; while (left < right && sl != t);
do sr += A[right--]; while (left < right && sr != t);
return left <= right && sl == sr;
}
};
范文参考:
先计算均值,再看是否能有足够的部分能够和为均值 - 将数组分成和相等的三个部分 - 力扣(LeetCode)
辅助数组sum,sum[i]存储前i个数的和(1013. 将数组分成和相等的三个部分) - 将数组分成和相等的三个部分 - 力扣(LeetCode)
辅助数组sum,sum[i]存储前i个数的和(1013. 将数组分成和相等的三个部分) - 将数组分成和相等的三个部分 - 力扣(LeetCode)