字符串问题：去掉字符串中连续出现 k 个 0 的子串

【题目】

　　给定一个字符串 str 和 一个整数 k, 如果 str 中正好有连续 k 个 ‘0’ 字符出现时，把 k 个连续的 ‘0’ 字符去除，返回处理后的字符串。

【举例】

　　str="A00B", k=2, 返回 “A  B”

　　str="A0000B000", k=3, 返回 “A0000B”

【难度】

　　一星

【解答】

public class Main {

    public static void main(String[] args) {
        System.out.println(new Main().removeKZeros("A00B", 2));//A  B
        System.out.println(new Main().removeKZeros("A0000B000", 3));//A0000B
    }
    
    public String removeKZeros(String str, int k){
        if(str == null || k < 1) return str;
        char[] chs = str.toCharArray();
        int count = 0;  //统计连续出现字符'0'的次数
        int start = -1; //记录开始出现字符'0'的位置
        for(int i = 0, len = chs.length; i < len; i++){
            char ch = chs[i];
            if(ch == '0'){
                count++;
                start = start == -1 ? i : start;
            }else{ //若遇到非'0'字符
                if(count == k){//表示出现连续 k 个 '0', 将字符 '0' 设置为空字符
                    while(count-- > 0){
                        chs[start++] = 0; //设置为空字符, ascii 码中 0 表示空字符（Null）
                    }
                }
                //重新统计
                count = 0;
                start = -1;
            }
        }
        //判断结尾是否出现 k 个 0' 字符
        if(count == k){
            while(count-- > 0){
                chs[start++] = 0;
            }
        }
        
        return String.valueOf(chs);
    }
    
}