URAL 1007 Code Words

#include<stdio.h>
#include<string.h>
#define MAXD 1010
int N, M, A[MAXD], S;
char b[MAXD];
void checkreplace()
{
int i, j, k;
for(i = 0; i < M; i ++)
if(b[i] == '1' && (S - i - 1) % (N + 1) == 0)
{
b[i] = '0';
printf("%s\n", b);
return ;
}
}
void checkremove()
{
int i, j, k;
for(i = 0; i < M; i ++)
{
if((S + A[i]) % (N + 1) == 0)
{
for(j = 0; j < i; j ++)
printf("%c", b[j]);
printf("0");
for(j = i; j < M; j ++)
printf("%c", b[j]);
printf("\n");
return ;
}
if((S + A[i] + i + 1) % (N + 1) == 0)
{
for(j = 0; j < i; j ++)
printf("%c", b[j]);
printf("1");
for(j = i; j < M; j ++)
printf("%c", b[j]);
printf("\n");
return ;
}
}
printf("%s", b);
if(S % (N + 1) == 0)
printf("0\n");
else
printf("1\n");
}
void checkinsert()
{
int i, j, k;
for(i = 0; i < M; i ++)
if((b[i] == '1' && (S - i - A[i]) % (N + 1) == 0) || (b[i] == '0' && (S - A[i]) % (N + 1) == 0))
{
b[i] = '\0';
printf("%s", b);
printf("%s\n", b + i + 1);
return ;
}
}
void solve()
{
int i, j, k;
M = strlen(b);
A[M] = S = 0;
for(i = M - 1; i >= 0; i --)
{
A[i] = A[i + 1] + b[i] - '0';
if(b[i] == '1')
S += i + 1;
}
if(N == M)
{
if(S % (N + 1) == 0)
printf("%s\n", b);
else
checkreplace();
}
else if(M < N)
checkremove();
else
checkinsert();
}
int main()
{
scanf("%d", &N);
while(scanf("%s", b) == 1)
{
solve();
}
return 0;
}