leetcode1573.分割字符串的方案数

最新推荐文章于 2026-04-02 14:57:08 发布

原创最新推荐文章于 2026-04-02 14:57:08 发布 · 556 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

leetcode算法题解

博客围绕二进制串分割问题展开。题目要求将二进制串 s 分割成 3 个非空字符串，使三者中字符 ‘1’ 的数目相同，求分割方案数。解题思路为，先统计 ‘1’ 的数量，若不是 3 的倍数则无法分割；若全是 0 用隔板法；反之确定每堆 ‘1’ 的数量，找出第一堆和第三堆的最小范围，统计可放隔板位置并计算方案数。

题目大意

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

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

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

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

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

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

解题思路

首先统计字符串中“1”的数量，如果不是3的倍数，在必定无法分割，返回0；
如果全是0，那就是排列组合里面的隔板法了（n个0，中间共有n-1个空，分成3堆，因此选择两个空进行分割即可）；
反之，则必定可以分割。先计算每堆有多少个1（假设x个）。则从前往后找，直到找到第x个1的位置，即为first；从后往前找，直到找到第x个1的位置，即为last；则0~first是第一堆的最小范围，last ~back是第三堆的最小范围。
从first+1位置往后数连续的0的个数，从last-1位置往前数连续的0的个数。这些位置是可以放隔板的位置，然后乘一下即可。

不过溢出太讨厌了。。。用了long long。。。

class Solution {
public:
    int numWays(string s) {
        int total = 0;
        for (char ss : s){
            if (ss =='1'){
                total++;
            }
        }
        if (total % 3 != 0) return 0;
        if (total == 0) return (s.size() - 1) * (s.size() - 2) / 2 % (1000000007);
        int oneStr = total / 3;

        int firstIndex = 0, lastIndex = s.size() - 1, tmp = 0;
        for (; firstIndex < s.size(); ++firstIndex){
            if (s[firstIndex] == '1'){
                tmp++;
            }
            if (tmp == oneStr){
                break;
            }
        }
        tmp = 0;
        for (; lastIndex >= 0; --lastIndex){
            if (s[lastIndex] == '1'){
                tmp++;
            }
            if (tmp == oneStr){
                break;
            }
        }
        int firstZeroNums = 0, lastZeroNums = 0;
        for (firstIndex = firstIndex + 1; firstIndex < s.size(); firstIndex++){
            if (s[firstIndex] == '0'){
                firstZeroNums++;
            }
            else{
                break;
            }
        }
        for (lastIndex = lastIndex - 1; lastIndex >= 0; lastIndex--){
            if (s[lastIndex] == '0'){
                lastZeroNums++;
            }
            else{
                break;
            }
        }
        return (long long)(firstZeroNums + 1)  * (long long)(lastZeroNums + 1) % (1000000007);
    }
};