L1-039 古风排版 Python

题目链接.

题目

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

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

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

输入样例：

4
This is a test case
结尾无空行

输出样例：

asa T
st ih
e tsi
ce s
结尾无空行

要点

• 注意字符串长度刚好可以被整除时
• 注意不能被整除时，需要补充空格
• 注意当N小于字符串长度时，需要注意不要超过索引

代码

# 正整数N
n = int(input())
# 非空字符串
s = input()

columns = 0
# 计算需要几列
if len(s) % n == 0:
    columns = len(s) // n - 1
else:
    columns = len(s) // n

# 循环输出结果
# 输出所有行的结果
for i in range(n):
    # 输出一行的元素
    for j in range(columns, -1, -1):
        # 如果最后一列不足N个，补充空格
        if len(s) <= j * n + i:
            print(" ", end='')
        else:
            print(s[j * n + i], end='')
    print()