leetcode 1013. 将数组分成和相等的三个部分

leetcode 1013. 将数组分成和相等的三个部分

题目详情

题目链接
给你一个整数数组 A，只有可以将其划分为三个和相等的非空部分时才返回 true，否则返回false。形式上，如果可以找出索引 i+1 < j 且满足 (A[0] + A[1] + … + A[i] == A[i+1] + A[i+2] + … + A[j-1] == A[j] + A[j-1] + … + A[A.length - 1]) 就可以将数组三等分。

• 示例 1：
  输出：[0,2,1,-6,6,-7,9,1,2,0,1]
  输出：true
  解释：0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1
• 示例 2：
  输入：[0,2,1,-6,6,7,9,-1,2,0,1]
  输出：false
• 示例 3：
  输入：[3,3,6,5,-2,2,5,1,-9,4]
  输出：true
  解释：3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4
• 提示：

1. 3 <= A.length <= 50000
2. -10^4 <= A[i] <= 10^4

我的代码

#include <vector>

using namespace std;

class Solution {
public:
    bool canThreePartsEqualSum(vector<int>& A) {
        int sum = 0;
        for (int a: A) {
            sum += a;
        }
        if (sum % 3 != 0) {
            return false;
        }
        int sum0 = sum / 3;
        int sum1 = 0;
        for (int i = 0; i < A.size(); ++i) {
            sum1 += A[i];
            if (sum1 != sum0) {
                continue;
            }
            int sum2 = 0;
            int j;
            for (j = i + 1; j < A.size(); ++j) {
                sum2 += A[j];
                if (sum2 != sum0) {
                    continue;
                }
                if (j == A.size() - 1) {
                    break;
                }
                int sum3 = 0;
                for (int k = j + 1; k < A.size(); ++k) {
                    sum3 += A[k];
                }
                if (sum3 == sum0) {
                    return true;
                }
            }
        }
        return false;
    }
};

我的成绩

执行结果：通过
执行用时 :44 ms, 在所有 C++ 提交中击败了98.98%的用户
内存消耗 :32.1 MB, 在所有 C++ 提交中击败了5.26%的用户

一些想法

1. 因为三部分和要相同，所以我们可以先求出总和，如果总和对三取余数，那么必定不满足题意。
2. 然后分别计算出综合除以三的索引即可。
3. leetcode的计算时间会有浮动。

执行用时为 24 ms 的范例

// An highlighted block
class Solution {
public:
    bool canThreePartsEqualSum(vector<int>& A) {
        int sum = 0, currSum = 0, k = 0;
        for(auto it : A)
            sum += it;
        if(sum % 3 != 0)
            return false;
        for(auto it : A)
        {
            currSum += it;
            if(currSum == sum / 3){
                currSum = 0;
                k++;
            }
        }
        if(currSum == sum / 3)
            k++;
        return k==3;           
    }
};

思考

1. 这个代码我测试时无法通过
2. 应该是判断条件少了一个。