求助！C语言

样例过了，但是全1.2s超时，能不能在优化了，求大佬解答

int main()
{
    int zz=1,a[40002]={0},n=0,count;
    char c,d='0';
    while((c=getchar())!='\n')
    {
        n++;
            if(c==d)
                a[zz]++;
            else
                a[++zz]++;
            d=c;
    }
    count=n;
    while(count!=n*n)
    {
        c=getchar();
        if(c=='0'||c=='1')
        {
            if(c==d)
                a[zz]++;
            else
                a[++zz]++;
            d=c;
            count++;
        }

    }
    printf("%d ",n);
    for(int i=1;i<zz;i++)
        printf("%d ",a[i]);
    printf("%d",a[zz]);

    return 0;
}