python字符串问题：字符串的展开

Question

这是一道OpenJudge和洛谷里都有的题目，我的解题思路是先考虑 '-' 两端的字符是否同为小写字母或同为数字，并且右端的ASCII码大于左端。然后考虑p1是否为3，最后再考虑p1和p3的具体取值。我的代码如下：

def fillstr(string, p1, p2, p3):
    s = [i for i in string]
    for i in range(len(s)):
        if s[i] == '-' and ord(s[i - 1]) < ord(s[i + 1]):
            if (s[i - 1].isdigit() and s[i + 1].isdigit()) or (s[i - 1].isalpha() and s[i + 1].isalpha()):
                if p1 != 3:
                    s[i] = [p2 * chr(x) for x in range(ord(s[i - 1]) + 1, ord(s[i + 1]))]
                elif p1 == 3:
                    s[i] = [p2 * (ord(s[i + 1]) - ord(s[i - 1]) - 1) * '*']
    finalstr = []    #在执行下面的函数之前，s里应该被替换的 ‘-’ 都以子列表的形式存储，所以下面的条件语句都只考虑子列表
    for i in s:
        if isinstance(i, list) and p1 == 2:
            i = [j.upper() for j in i]
        if isinstance(i, list) and p3 == 1:
            finalstr.extend(i)
        elif isinstance(i, list) and p3 == 2:
            finalstr.extend(i[::-1])
        else:
            finalstr.append(i)
    return ''.join(finalstr)


para_list = [int(i) for i in input().split()]
String = input()
print(fillstr(String, para_list[0], para_list[1], para_list[2]))

这串代码在洛谷的评分中是满分，但OpenJudge的评分结果是Runtime Error，请问是因为还应该再进一步简化吗？

Answer 1

• 您的解题思路是：
  1、考虑"两端的字符是否同为小写字母或同为数字，并且右端的ASCII码大于左端；
  2、考虑p1是否为3；
  3、考虑p1和p3的具体取值。

• $我的审题$
  参数说明：
   
  p1(1.填充小写字母；2.填充大写字母；3.填充*号。数字在1、2情况都填充数字。)
   
  p2(填充字符重复个数)
  p3(1.正序；2.逆序)

• 填充规则
   
  -号端是相邻的大-小字符，去除-号；前大后小，保持原状。_{(我认为，据题意如有p1不在1~3的范围或者p2不是1、2时，也应该是保持原状的。)}

• 看您是用列表结构数据来处理的，python中的列表是不高效的，您试着不用列表试试。比如您代码的第一行可以注释掉，python中的字符串str也可以可以直接遍历的序列。

• 代码运行效果截屏图片
  

  

python代码

def fillRange(s, *p): # 用*号收集p1、p2、p3。
    
    # 首先剔除不用展开的情形，省点time。

    if p[0] not in (1, 2, 3) or p[-1] not in (1, 2):
        print(f"\n{' p参数错误！':~^35}")
        return
    elif '-' not in s:
        print(f"\n{' 字符串格式错误！':~^32}")
        return
    
    #用str.split('-')拆分字符串，减少遍历次数。
    s = s.split('-')
    result = s[0]
    
    for i in range(1, len(s)): # 遍历到倒数第二组子字符串。
        a = result[-1] # -前字符。
        b = s[i][0] # -后字符
        a2, b2 = ord(a), ord(b) # a、b的ASCII码。
        if a.isdigit() and b.isdigit() and b > a:
            chars = '' if b2-a2==1 else char(a2, b2, p)
        elif a.islower() and b.lower() and b > a: # 小写字母可以直接比较大小，实则是比较ASCII编码。
            chars = '' if b2-a2==1 else char(a2, b2, p)
        else:
            chars = '-'
        
        result = ''.join([result, chars, s[i]])

    return result


def char(a, b, p):
    ''' 生成连续字符串 '''
    a += 1
    
    if p[0] == 3:
        chars = ''.join(['*'*p[1] for i in range(a, b)])
    else:
        chars = ''.join([chr(i)*p[1] for i in range(a, b)])

    if p[-1] == 2:
        return chars[::-1]

    return chars


if __name__ == '__main__':
    s = input(f"\n输入字符串：").strip()
    while 1:
        p = list(map(int, input(f"输入字p参数(如1 2 1)：").strip().split()))
        if len(p)==3:
            break 
        
        print(f"{' 重新输入！':~^35}\n")
    print(f"\n输出：{fillRange(s, *p)}")
    

试试看，这样子还超时不？我的代码解析说明，请看秒后整理的学习笔记。

---