LeetCode每日一题：927. 三等分 （困难） 数组 数学

题目：力扣

给定一个由 0 和 1 组成的数组 arr ，将数组分成  3 个非空的部分 ，使得所有这些部分表示相同的二进制值。

如果可以做到，请返回任何 [i, j]，其中 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, -1]。

注意，在考虑每个部分所表示的二进制时，应当将其看作一个整体。例如，[1,1,0] 表示十进制中的 6，而不会是 3。此外，前导零也是被允许的，所以 [0,1,1] 和 [1,1] 表示相同的值。

示例 1：

输入：arr = [1,0,1,0,1]
输出：[0,3]
示例 2：

输入：arr = [1,1,0,1,1]
输出：[-1,-1]
示例 3:

输入：arr = [1,1,0,0,1]
输出：[0,2]
 

提示：

3 <= arr.length <= 3 * 104
arr[i] 是 0 或 1

思路：1、求出数组中1的个数，如果不够三等分则直接结束

2、如果全都是0则答案可以为【0，2】也可以为【0，arr.size()-1】

3、从后往前求出最后一等分的0的个数k，k为每一等分后导0

4、根据1和0的个数找出答案的下标

5、计算出三等分的三个数的值是否相等

官方题解：力扣

class Solution {
public:
    vector threeEqualParts(vector& arr) {
        vector arr2;
        int cnt = 0;
        for(auto it : arr){
            if(it==1) cnt ++;
        }
        if(cnt % 3 && cnt) return {-1 , -1};
        if(!cnt) return {0,(int)arr.size()-1};
        cnt /=3;
        arr2 = arr;
        int k =0;
        while(!arr2.back())  {
            k++;
            arr2.pop_back();
        }
       // cout<