P1320 压缩技术（续集版），原题如下：

题目描述

设某汉字由N X N的0和1的点阵图案组成，如下图。我们依照以下规则生成压缩码。连续一组数值：从汉字点阵图案的第一行第一个符号开始计算，按书写顺序从上到下，由左到右。第一个数表示连续有几个0，第二个数表示接下来连续有几个1，第三个数再接下来连续有几个0，第四个数接着连续几个1，以此类推。。。

例如: 以下汉字点阵图案：

0001000

0001000

0001111

0001000

0001000

0001000

1111111

对应的压缩码是： 7 3 1 6 1 6 4 3 1 6 1 6 1 3 7 （第一个数是N ,其余各位表示交替表示0和1 的个数，压缩码保证 N X N=交替的各位数之和）

输入格式

汉字点阵图（点阵符号之间不留空格）。（3<=N<=200）

输出格式

一行，压缩码。

根据题意编写代码如下：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define STRING_SIZE 40001

int main()
{
    char str[STRING_SIZE], ch, *p, *pe;
    int n, i, key = 0, num = 0;
    pe = &str[0];
    while ((ch = getchar()) != '\n') {
        *pe++ = ch;
    }
    n = pe - str;
    for (i = 1; i < n; ++i) {
        while ((ch = getchar()) != '\n') {
            *pe++ = ch;
        }
    }
    printf("%d", n);
    for (p = &str[0]; p < pe; ++p) {
        if ((*p) == (char)(key + '0'))
            ++num;
        else {
            printf(" %d", num);
            key = !key;
            num = 1;
        }
    }
    printf(" %d", num);

    return 0;
}

代码在Code::Blocks上运行正常，用题目中的输入数据和测试数据1都能输出正确的结果，但在洛谷上提交后总是报Runtime error和Wrong answer，我反复查看也不知错在何处，请大侠指教！