去掉字符串中连续出现k个0的字串

A000B0000  K=3    ----->     AB0000

这里注意一下：

当循环结束后，要检查下count是否等于k,如果不等于k,要把s后面的补回去

//
//  main.cpp
//  去掉字符串中连续出现k个0的字串
//
//  Created by zjl on 16/8/13.
//  Copyright © 2016年 zjl. All rights reserved.
//  A0000B000     A0000B

#include <iostream>
#include <string>
using namespace std;

void solve(string& s, int k){
    int pos = -1, count = 0;
    int len = s.size();
    for(int i = 0; i < s.size(); i++){
        if(s[i] == '0')
            count++;
        else{
            if(count == k || count == 0){
                s[++pos] = s[i];
            }else{
                while(count){
                    s[++pos] = s[i - count];
                    count--;
                }
                s[++pos] = s[i];
            }
            count = 0;
        }
    }
    if(count != k && count > 0){
            while(count--)
            s[++pos] = s[ len - 1 - count];
    }

    s = s.substr(0, pos+1);
}

int main(int argc, const char * argv[]) {
    // insert code here...
    string s = "A0000B0000";
    solve(s, 3);
    cout << s<<endl;
    return 0;
}