Leetcode：3258

1，题目

2，思路

• 遍历每个元素，以每个元素为起点计算满足条件（满足内部1或0不超过k值）
• 每次开始时判断设置初始值 one与zero为0次
• 初始判断头元素是1或0这次需计算到条件里
• 内循环条件不满足时结束，不需要继续执行后面循环

3，代码

class Solution3258 {
    public int countKConstraintSubstrings(String s, int k) {
        if (k > 0) {
            int count = s.length();//记录满足要求子串 初始为长度为1的子串直接记录
            int one, zero;
            String test;
            for (int i = 0; i < s.length(); i++) {//每个元素为子串开头计算
                one = zero = 0;
                if (s.charAt(i) == '1') {//当前子串元素判断
                    one = 1;
                } else zero = 1;
                for (int j = i + 1; j < s.length(); j++) {//第二个元素开始往后判断
                    if (s.charAt(j) == '1') {//后续元素等于‘1’
                        one++;
                        if (one <= k || zero <= k)//满足一个条件就++
                            count++;
                        else break;
                    } else if (s.charAt(j) == '0') {//后续元素等于‘0’
                        zero++;
                        if (one <= k || zero <= k)//满足一个条件就++
                            count++;
                        else break;
                    }
                }
            }
            return count;
        }
        return 0;
    }
}