去掉连续出现k次的0

#include <iostream>
#include <string>

using namespace std;

void remove(string inistring,int k)
{
    int length=inistring.length();
    int count=0;//0的个数
    string s;
    int pos=0;//s字符串中下一个要插入位置
    int num;//处理后0的个数
    for(int i=0;i<length;++i)
    {
        char c=inistring[i];
        if(c=='0')
        {
            count++;
        }
        else
        {
            num=count%k;
            s.insert(pos,num,'0');
            pos+=num;

            s.insert(pos,1,c);
            pos++;
            count=0;
        }
    }
    num=count%k;
    s.insert(pos,num,'0');
    cout<<s<<"\n";
}

int main()
{
    string str="A00000B00000";
    int k;
    cin>>k;
    remove(str,k);
    return 0;
}