蓝桥杯的字母图形题目

问题描述
利用字母可以组成一些美丽的图形，下面给出了一个例子：

ABCDEFG

BABCDEF

CBABCDE

DCBABCD

EDCBABC

这是一个5行7列的图形，请找出这个图形的规律，并输出一个n行m列的图形。

输入格式
输入一行，包含两个整数n和m，分别表示你要输出的图形的行数的列数。
输出格式
输出n行，每个m个字符，为你的图形。
样例输入
5 7
样例输出
ABCDEFG
BABCDEF
CBABCDE
DCBABCD
EDCBABC

这题的思路主要是找出每行的递归式，
设m=26，则第一行应该为：“ABCDEFG…XYZ"，所以，m=y时，第一行为：
”ABCDEFG…XYZ"[:y]的切片
已知，m=7时，第一行输出’‘ABCDEFG‘’
，那么第二行就输出“BABCDEF”，观察发现可知，
第二行为： B+第一行[:m-1]
所以代码如下

刚开始我的代码是这样的

def cut(n,m):
    s='ABCDEFGHIJKLMNOPQRSTUVWXYZ'
    index=0
    p="ABCDEFGHIJKLMNOPQRSTUVWXYZ"[:m]
    while True:
        yield p
        index+=1
        p=s[index]+p[:m-1]
        if index==n:
            break
def run():
    n,m=list(map(int,input().strip().split()))
    for i in cut(n,m):
        print(i)
if __name__=='__main__':
    run()

然后到练习系统中测试发现只得了80 分，出现了代码错误的提示，这样我认为是出现了异常

用了
26 26
这样的极端数据测试
果然发现了

Traceback (most recent call last):
  File "/Users/mac/字母图形.py", line 16, in <module>
    run()
  File "/Users/mac/字母图形.py", line 13, in run
    for i in cut(n,m):
  File "/Users/mac/字母图形.py", line 8, in cut
    p=s[index]+p[:m-1]
IndexError: **string index out of range`**

出现了这样的异常，所以我将代码微微调整了一下
if判断在前，得：

def cut(n,m):
    s='ABCDEFGHIJKLMNOPQRSTUVWXYZ'
    index=0
    p="ABCDEFGHIJKLMNOPQRSTUVWXYZ"[:m]
    while True:
        yield p
        index+=1
        if index==n:
            break
        p=s[index]+p[:m-1]
        
def run():
    n,m=list(map(int,input().strip().split()))
    for i in cut(n,m):
        print(i)
if __name__=='__main__':
    run()

这样就没报异常了，然后放到系统里面提交，成功通过！