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

题目：

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

举例：

str="A00B"，k=2，返回"A002"；

str="A0000B000"，k=3，返回"A0000B"。

难度：★

思路：

对于处理字符串各种问题，以前我都是以双重嵌套著称，时间复杂度就可想而知了。看了这本书以后，根据别人的思路，再加上有意识地往单循环思考怎么处理。慢慢地，也就养成习惯了，加油吧，前面的路还很长，不能任性，继续加油！！！（扯远了。。。）

仍是从左往右遍历整个字符串，判断每个字符是否为'0'，若是'0'，则对count加一处理；若不是'0'，则判断count和k是否相等，若相等则只将当前位字符加入新字符串中；若不等则将前面几位加上当前位的字符串加入新字符串中。

代码：

#include <iostream>#include <string>using namespace std;int main(){string str;getline(cin, str);int k;cin >> k;int len = str.size();int i, count = 0;string new_str;if (k == 0 || len == 0){cout << str;return 0;}for (i = 0; i < len; i++){if (str[i] == '0')count++;else{if (count != k)new_str += str.substr(i - count, count+1);elsenew_str += str[i];count = 0;}}cout << new_str << endl;system("pause");return 0;}