LeetCode每日一题（2020/3/11）

LeetCode每日一题（2020/3/11）

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

提示：
3 <= A.length <= 50000
-10^4 <= A[i] <= 10^4

方法1：
1.判断数组和是否是3的整数倍，如果不是证明无法将其划分为3个相等的非空部分，返回false；
2.将数组和的1/3记做part_sum，如果可以将其划分为3个相等的非空部分，则每部分的和应为part_sum；
3.定义当前数组和s与标记f，遍历数组，当s等于part_sum时，s清零，f加一；
4.当遍历完当前数组时，如果f小于3，返回false，否则返回true。
python3：

class Solution:
    def canThreePartsEqualSum(self, A: List[int]) -> bool:
        all_sum = sum(A)
        if(all_sum%3!=0): return False
        part_sum = all_sum/3
        i, s, f = 0, 0, 0
        while(i<len(A)):
            s += A[i]
            i += 1
            if(s==part_sum):
                s = 0
                f += 1
        if f<3:
            return False
        else:
            return True

C++:

class Solution {
public:
    bool canThreePartsEqualSum(vector<int>& A) {
        int all_sum = 0;
        for(int j=0;j<A.size();j++){
            all_sum += A[j];
        }
        if(all_sum%3!=0) return false;
        int part_sum = all_sum / 3;
        int i=0,s=0,f=0;
        while(i<A.size()){
            s+=A[i];
            if(s==part_sum){
                s=0;
                f++;
            }
            i++;
        }
        if(f<3) return false;
        else return true;
    }
};

方法2：双指针。整体思路与方法1相同，不同之处为使用双指针在数组两端开始检测，只有两端都检测到和为part_sum时，并且left+1<right时，可以保证数组可以被分成三个和相同的子数组。
C++：

class Solution {
public:
    bool canThreePartsEqualSum(vector<int>& A) {
        int all_sum = 0;
        for(int j=0;j<A.size();j++){
            all_sum += A[j];
        }
        if(all_sum%3!=0) return false;
        int part_sum = all_sum / 3;
        int left=0, right=A.size()-1;
        int left_sum=A[left], right_sum=A[right];
        while(left+1<right){
            if(left_sum==part_sum && right_sum==part_sum) return true;
            if (left_sum!=part_sum) left_sum+=A[++left];
            if (right_sum!=part_sum) right_sum+=A[--right];
        }
        return false;
    }
};

python3:

class Solution:
    def canThreePartsEqualSum(self, A: List[int]) -> bool:
        all_sum = sum(A)
        if(all_sum%3!=0): return False
        part_sum = all_sum/3
        left, right = 0, len(A)-1
        left_sum, right_sum = A[left], A[right]
        while(left+1<right):
            if left_sum==part_sum and right_sum==part_sum: return True
            if left_sum != part_sum:
                left += 1
                left_sum += A[left]
            if right_sum != part_sum:
                right -= 1
                right_sum += A[right]
        return False