【携程秋招】230907四、好串数量 ＜模拟＞

【携程秋招】230907四、好串数量

有一个只包含‘0’和‘1’的字符串，想知道这个字符串有多少个好子串？
一个字符串如果是好串，那么该字符串的所有前缀，‘0’的数量严格大于‘1’的数量。

输入：
一个只含‘0’和‘1’的字符串，长度不超过100000。

输出：
一个整数，代表答案。

输入：

100

输出

3

说明

子区间[2,2]、[2,3]、[3,3]组成的子串都是一个好串。

输入：

10010

输出

6

说明

子区间[2,2]、[2,3]、[3,3]、[2,4]、[2,5]、[5,5]组成的子串都是一个好串。

题解

举例说明：
例如0011100
应该为1+2+1+0+0+1+2=7
由等式找规律

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String str = in.next();

        long curSum = 0;
        long count = 0;
        for (int i = 0; i < str.length(); i++) {
            //是0
            if (str.charAt(i) == '0') {
                curSum++;
            }
            //是1
            else {
                curSum--;
            }

            if (curSum >= 1) {
                count += curSum;
            } else {
                curSum = 0;
            }
        }
        System.out.println(count);

    }
}