Leetcode927 Three Equal parts

题目大意

原题链接：https://leetcode.com/problems/three-equal-parts/
题目大意：

• 考察一个给定的字符串是否能分成表示二进制的值相等的三部分
• 忽略前导0，但后面的0不能忽略
• 若不能划分，则返回 [-1,-1]

我的思路

• 计算1的个数就可以大致划分字符串，若1的个数不能被3整除，那么一定不能成功划分（必至少有一组的1的个数与其他不一样，大小也必定不一样）
• 若1的个数为0，则必定可以成功划分
• 若1的个数不为0（至少是3），假设每组有n个1，用标记标记每组第一个1出现的位置，开始匹配。
• 若有不等的情况匹配失败，否则成功

AC代码

#include <iostream>

using namespace std;

class Solution {
public:
    vector<int> threeEqualParts(vector<int>& A) {
        vector<int>rel;
        int countn=0;
        for(int i=0;i<A.size();i++)
        {
            if(A[i]==1){countn++;}
        }
        if(countn%3!=0){rel={-1,-1};return rel;}
        int num=countn/3;
        int sign1=-1,sign2=-1,sign3=-1;
        int tmpcount=0;
        if(num==0){rel={0,2};return rel;}
        for(int i=0;i<A.size();i++)
        {
            if(A[i]==1){
                tmpcount++;
                if(tmpcount==1){sign1=i;}
                else if(tmpcount==num+1){sign2=i;}
                else if(tmpcount==num*2+1){sign3=i;}
            }
        }
        int s1=sign1,s2=sign2,s3=sign3;
        for(s3=sign3;s3<A.size();s3++)
        {
            if(!(A[s1]==A[s2]&&A[s3]==A[s2])){break;}
            else{
                s1++;
                s2++;
            }
        }
        if(s3<A.size()){return {-1,-1};}
        rel={s1-1,s2};
        return rel;
    }
};