Pythonを使ったプログラミングの問題について教えてください。

Question

[問題]

1. 相性をチェックする二人の名前を並べた英小文字からなる文字列を入力します。
2. "a" を 1、"b" を 2、...、"z" を 26 として、文字列を数列に変換します。この数列を A とします。
3. 数列 A の隣り合う 2 つの数を足して前から順番に並べた新しい数列 A' を作り、これを新たに A とします。このとき、A の要素の大きさが 101 を超えていた場合、その要素から 101 を引きます。
4. 数列 A の要素数が 1 になるまで 3. の手順を繰り返します。A の要素数が 1 となったとき、残った要素の値を「二人の相性」とします。

名前の並べ方は 2 通りあります。そこで、あなたは相性占いの結果として、 2 通りの方法で計算した「二人の相性」のうち大きい方を出力するようにプログラムを組むことにしました。

たとえば、ai さんと ueo さんの名前を "aiueo" として並べたとき、「二人の相性」は 97 になります。
[1, 9, 21, 5, 15]
[10, 30, 26, 20]
[40, 56, 46]
[96, 1]
[97]

ueo さんの名前を先に持ってきて "ueoai" とすると、「二人の相性」は 43 になります。
[21, 5, 15, 1, 9]
[26, 20, 16, 10]
[46, 36, 26]
[82, 62]
[43]

相性占いをする二人の人物の名前が与えられたとき、相性占いの結果を出力するプログラムを作成してください。

a , b = input().split()
abc = {'a':1,'b':2,'c':3,'d':4,'e':5,'f':6,'g':7,'h':8,'i':9,'j':10,'k':11,'l':12,'m':13,'n':14,'o':15,'p':16,'q':17,'r':18,'s':19,'t':20,'u':21,'v':22,'w':23,'x':24,'y':25,'z':26}
c = a+b
l = []

for i in range(len(c)):

    x = c[i]

    y = abc[x]

    l.append(y)


print(l)

m = []
count = 0

for i in range(len(c)-1):
    count = l[i] + l[i+1]
    if count > 101:
        count -= 101
    m.append(count)

print(m)

n = []
p = 0

for i in range(len(c)-2):
    p = m[i] + m[i+1]
    if p > 101:
        p -= 101
    n.append(p)

print(n)

q = []
j = 0

for i in range(len(c)-3):
    j = n[i] + n[i+1]
    if j > 101:
        j -= 101
    q.append(j)

print(q)

v = []
h = 0
for i in range(len(c)-4):
    h = q[i] + q[i+1]
    if h > 101:
        h -= 101
    v.append(h)

print(v)

上のコードの通り、5文字までは実行でき解答を求めることができるのですが、a b の標準入力の範囲が1 ≦ (a の長さ) ≦ 1,000、1 ≦ (b の長さ) ≦ 1,000の時のプログラムの書き方が思いつきません。
np.cumsumを使って解くのかなと思ったのですが、できませんでした。
プログラミングに詳しい方、
入力例　alice bob
出力　97
などの上記の範囲（1 ≦ (a の長さ) ≦ 1,000、1 ≦ (b の長さ) ≦ 1,000）に対応するプログラムの書き方を教えてください。

Answer 1

提示コードで繰り返し書いている部分をリストの要素が１個になるまでといったループにおさめればよいです。

def calc(s):
    cur = [ ord(c)-ord('a')+1 for c in s] # 数値リストに変換

    # リストの要素が１個になるまで
    while( len(cur) > 1):
        next = []
        for i in range(len(cur)-1):
            v = cur[i] + cur[i+1]
            if v > 101:
                v -= 101
            next.append(v)
        cur = next
    return cur[0]


a, b= 'ai', 'ueo'

print(calc(a+b)) # 97
print(calc(b+a)) # 43

Comments

[zero1132]

回答ありがとうございます。答えを導くことができたのですが、私の知識不足で理解できない部分がありますので、ご教授ください。
cur = [ ord(c)-ord('a')+1 for c in s] # 数値リストに変換のところで、
aiueo[1, 9, 21, 5, 15]
ueoai[21, 5, 15, 1, 9]となっていると思うのですが、ord(c)-ord('a')+1の部分を教えてください。

[8524ba23]

ord関数については調べてみましたでしょうか？

[zero1132]

調べてみました。print(ord('a'))の時は97と出力されるみたいなのですが、今回の場合ではaの時に1になるのがよく分かりませんでした。

[8524ba23]

「ord('a') - ord('a') + 1」で「a」の時は１になります。
「ord('b')」は98なので同じように2になります。

[zero1132]

なるほど、そういうことだったんですね。
理解でました。有り難うございました！