LeetCode 1573. 分割字符串的方案数（数学、字符串）

最新推荐文章于 2024-05-20 21:56:33 发布

岁忧

最新推荐文章于 2024-05-20 21:56:33 发布

阅读量296

点赞数 11

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展 c++ 数学

本文链接：https://blog.csdn.net/weixin_46028214/article/details/138747857

版权

LeetCode 专栏收录该内容

228 篇文章 0 订阅

订阅专栏

1573. 分割字符串的方案数

在这里插入图片描述

思路：先遍历一遍字符串，统计有多少个‘1’，如果为sum=0，那么就是C（n-1，2）；如果sum%3！=0，那么为0；接下来就是统计两个分界线可以放在哪些位置，对于第一条分界线，其左边的1数量一定要为sum/3；对于第二条分界线，其右边的1数量一定也为sum/3，这样我们看看两条分界线在两边分别有多少种落脚点left、right，然后得到其组合数left*right即可。

class Solution {
public:
    int numWays(string s) {
        int mod=1e9+7;
        int n=s.size();
        int sum=0;
        for(int i=0;i<n;i++){
            if(s[i]=='1') sum++;
        }
        if(sum==0){
            long t=(long)(n-1)*(n-2)/2;
            return (int)(t%mod);
        }else if(sum%3!=0) return 0;
        sum/=3;
        int id1=-1,id2=-1;
        int ct=sum;
        int left=0;
        for(int i=0;i<n;i++){
            if(s[i]=='1'){
                if(ct){
                    ct--;
                }else{
                    left++;
                    break;
                }
            }else if(!ct){
                left++;
            }
        }
        ct=sum;
        int right=0;
        for(int i=n-1;i>=0;i--){
            if(s[i]=='1'){
                if(ct){
                    ct--;
                }else{
                    right++;
                    break;
                }
            }else if(!ct){
                right++;
            }
        }
        long t=(long)(left)*right;
        return (int)(t%mod);
    }
};