PTA天梯赛Python7-52 古风排版

中国的古人写文字，是从右向左竖向排版的。本题就请你编写程序，把一段文字按古风排版。

输入格式：

输入在第一行给出一个正整数N（<100），是每一列的字符数。第二行给出一个长度不超过1000的非空字符串，以回车结束。

输出格式：

按古风格式排版给定的字符串，每列N个字符（除了最后一列可能不足N个）。

输入样例：

4
This is a test case

输出样例：

asa T
st ih
e tsi
 ce s

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

代码实现：

n=int(input())
# n1用于后续循环中逐步减少步长
n1=n
# cnt用于打印换行
cnt=0
input_str=input()
# 如果字符串的长度不是 n 的倍数，在字符串末尾添加空格，使其长度成为 n 的倍数
if(len(input_str)%n !=0):
    input_str=input_str + (n-len(input_str)%n)* " "
for i in range(n):
    # 从字符串的末尾开始，向前每隔 n 个字符取一个字符。
    for j in range(len(input_str)-n1,-1,-n):
        print(f"{input_str[j]}",end="")
        cnt+=1
        if cnt%int(len(input_str)/n)==0:
            print()
    n1-=1