leetcode 927. 三等分(脑洞题)

xiaoxiao2025-10-25  18

题目:

给一串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(); } };

 

转载请注明原文地址: https://www.6miu.com/read-5038497.html

最新回复(0)