leetcode 927. 三等分（脑洞题）

题目：

给一串01串，然后从中分开成三份，然后问有没有可能组成三个二进制是相同的。

思路：

看大佬的思路才做出来的。。。。一开始我都想用kmp了。。。。

结果，二进制相同那么肯定1的个数是相同的。那么可以根据1的个数划分一个大概的界限。

有了每个区间1的个数后，我们可以计算出：第一个区间的起点firone，第二个区间的起点firtwo，第三个区间的起点firthree。

三个区间的最后一个1的位置：lastone，lasttwo

然后我们还知道要二进制数相同的话，那么结尾的0的个数肯定是要相同的，结尾0的个数我们已经知道了，从最后一个区间可以算出来。即为lastz

然后我们要算的就只有【firone，lastone+lastz】【firtwo，lasttwo+lastz】【firtwo，结尾】这三个区间是不是相同的。

注意全为0的情况。

代码：

class Solution {
public:
vector<int> fail(){
    vector<int>ans;
    ans.push_back(-1);
    ans.push_back(-1);
    return ans;
}

bool pan(vector<int>& A,int a,int b,int c,int len) //计算三个区间是不是相同
{
    for(int i=0;i<len;i++){
        if(A[a+i]!=A[i+b]||A[i+b]!=A[i+c])return false;
    }
    return true;
}


vector<int> threeEqualParts(vector<int>& A) {
int sz = A.size();
int num = 0;
for(int i=0;i<sz;i++){
    if(A[i])num++;
}
vector<int>ans;
    if(num%3!=0){
        return fail();
    }
    if(num==0){ //全0
        ans.push_back(0);
        ans.push_back(2);
        return ans;
        
    }
    num = num/3;
    int lastone=-1,lasttwo=-1,firtwo=-1,firthree=-1,firstone = -1,lastz = -1;
    int ji = 0;
    for(int i=0;i<sz;i++){
        if(firstone==-1&&A[i])firstone = i;
        if(A[i])ji++;
        if(ji==num&&lastone==-1)lastone = i;
        if(ji==num+1&&firtwo==-1)firtwo = i;
        if(ji==2*num&&lasttwo==-1)lasttwo = i;
        if(ji==2*num+1&&firthree==-1)firthree = i;
        if(ji==3*num)lastz++;
    }
   cout<<firstone<<" "<<lastone<<" "<<firtwo<<"  "<<lasttwo<<" "<<firthree<<" "<<lastz<<endl;
    if(lastone+lastz>=firtwo||lasttwo+lastz>=firthree)return fail(); //不满足后缀0
    if(pan(A,firstone,firtwo,firthree,sz-firthree)){
        ans.push_back(lastone+lastz);
        ans.push_back(lasttwo+1+lastz);
        return ans;
    }
    else return fail();
}

};