PTA天梯赛L1-039 古风排版 (20 分)疑问

Question

问题相关代码，请勿粘贴截图

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<stack>
#include<map>
using namespace std;
int main()
{
    int n,k=-1;
    char a[1010][1010];
    memset(a,' ', sizeof(a));
    string s;
    cin>>n;
    getchar();
    getline(cin,s);
    int len=s.length(),key=0;
    for(int i=1;;i++){
        for(int j=1;j<=n;j++){
            a[i][j]=s[++k];
            if(k==len)break;
        }
        if(k==len){
            key=i;
            break;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=key;j>=1;j--){
            cout<<a[j][i];
        }
        cout<<endl;
    }
    return 0;
}

运行结果及报错内容

样例：
4
This is a test case

样例在我编译器输出的是正确答案，在pta编译器上输出少了最后两行，调试了好久，都是如此，希望朋友们能解答下

Answer 1

k==len时s字符串越界了,多遍历了一个s字符串结束符'\0'
这样输出的第4行第一个是\0字符
asa T
st ih
e tsi
(\0字符)ce s
在pta编译器上是以文件的形式输出,\0表示文件结尾,\0之后的内容就不显示了
把两处if(k==len) 改成 if(k==len-1) 即可

你题目的解答代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<stack>
#include<map>
using namespace std;
int main()
{
    int n,k=-1;
    char a[1010][1010];
    memset(a,' ', sizeof(a));
    string s;
    cin>>n;
    getchar();
    getline(cin,s);
    int len=s.length(),key=0;
    for(int i=1;;i++){
        for(int j=1;j<=n;j++){
            a[i][j]=s[++k];
            if(k==len-1)break;  //改成k==len-1
        }
        if(k==len-1){  //改成k==len-1
            key=i;
            break;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=key;j>=1;j--){
            cout<<a[j][i];
        }
        cout<<endl;
    }
    return 0;
}

如有帮助，请点击我的回答下方的【采纳该答案】按钮帮忙采纳下，谢谢!