Leetcode 1573. 分割字符串的方案数(终于解决)

给你一个二进制串 s （一个只包含 0 和 1 的字符串），我们可以将 s 分割成 3 个 非空 字符串 s1, s2, s3 （s1 + s2 + s3 = s）。

请你返回分割 s 的方案数，满足 s1，s2 和 s3 中字符 '1' 的数目相同。

由于答案可能很大，请将它对 10^9 + 7 取余后返回。

示例 1：

输入：s = "10101"
输出：4
解释：总共有 4 种方法将 s 分割成含有 '1' 数目相同的三个子字符串。
"1|010|1"
"1|01|01"
"10|10|1"
"10|1|01"

示例 2：

输入：s = "1001"
输出：0

示例 3：

输入：s = "0000"
输出：3
解释：总共有 3 种分割 s 的方法。
"0|0|00"
"0|00|0"
"00|0|0"

示例 4：

输入：s = "100100010100110"
输出：12
 

提示：

• s[i] == '0' 或者 s[i] == '1'
• 3 <= s.length <= 10^5

Code：

class Solution {
public:
    int numWays(string s) {
        int res=0;
        int cnt=0;
        for(int i=0;i<s.length();i++)
        {
            if(s[i]=='1')
                cnt++;
        }
        if(cnt>0)
        {
            if(cnt%3)
                return 0;
        }
        else//这边是考虑全零（相当于求1到字符串长度-2的累加值）
        {
            uint64_t res=0;
            for(int i=1;i<=s.length()-2;i++)
                res+=i;
            return (res)%1000000007;
        }
        int num=cnt/3;
        uint64_t left=0;
        uint64_t right=0;
        for(int i=0;i<s.length();i++)
        {
            if(s[i]=='1')
            {
                left++;
            }
            if(left==num)
            {
                left=i;
                break;
            }
        }
        
        
        for(int i=s.length()-1;i>=0;i--)
        {
            if(s[i]=='1')
                right++;
            if(right==num)
            {
                right=i;
                break;
            }
        }
        uint64_t left2=0;
        uint64_t right2=0;
        for(int i=left+1;i<s.length();i++)
        {
            if(s[i]=='1')
            {
                left2=i;
                break;
            }
            
            
        }
        
        for(int i=right-1;i>=0;i--)
        {
            if(s[i]=='1')
            {
                right2=i;
                break;
            }
            
            
            
        }
        int exp=1000000007;
        return ((left2-left)%exp)*((right-right2)%exp)%exp;
    }
};