求大佬帮助

#include<bits/stdc++.h>
using namespace std;
int ans;
char a[205];
char z[40000];
int main()
{
    gets(a);
    int n=strlen(a);

    for(int i=n;i<n*n;i++)
    {
        cin>>z[i];
    }
    for(int i=0;i<n;i++)
    z[i]=a[i];
    cout<<n-1<<" ";
    if(z[0]!='0')cout<<0<<" ";
    for(int i=0;i<n*n;i++)
    {
        if(z[i]=='0')
        {
            ans++;
            if(z[i+1]=='1')
            {
                cout<<ans<<" ";
                ans=0;
            }
        }
        if(z[i]=='1')
        {
            ans++;
            if(z[i+1]=='0')
            {
                cout<<ans<<" ";
                ans=0;
            }
        }
    }
    if(ans!=0)cout<<ans;
    return 0;
}