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<int> threeEqualParts(vector<int>& arr) {
        vector<int> 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<<cnt << k <<endl;
        int tmp = 0 , tp = 0 ;
        int ans[3] ;
        for(int i = 0 ,j = 0; i < arr.size() ; i++)
        {
            if(arr[i] && tmp < cnt) {
                tmp ++;
            }
            if(!arr[i] && tmp == cnt && tp < k){
                tp ++;
            }
            if( tmp == cnt && tp == k){
                ans[j++] = i;
                tmp = 0 , tp = 0;
            }
        }
        //cout<<ans[0] <<" "<< ans[1] <<endl;
        unsigned int dig[3]={0};
        for(int i = 0 , j = 0; i < arr.size() ; i++ ){
            dig[j] = dig[j]<<1 | arr[i];
//因为 C++ 中对负数的左移没有定义，需要注意定义进位的类型要使用无符号数，这时就可以正确左移。
            if(i == ans[0] || i == ans[1]) j++;
        }
        ans[1]++;
        if( dig[0] == dig[1] && dig[1] == dig[2]){}
        else ans[0]=-1,ans[1]=-1;
        //cout<<dig[0] <<" " << dig[1] <<" " << dig[2] <<endl;
        return {ans[0] ,ans[1]};
    }
};