mor*_*ify -1 python recursion loops nested
我需要在Python中使用递归重写这段可怕的代码.嵌套的深度应该取决于函数rec的参数,但事实上我希望它是变量"a"的长度,即字符串.我很感激任何回应和线索如何处理这个问题.
def rec():
count=0
for i in range(len(letters)):
for j in range(i+1, len(letters)):
if letters[i]+letters[j] in a:
for k in range(j+1, len(letters)):
if letters[i]+letters[j]+letters[k] in a:
if letters[i]+letters[j]+letters[k]==a:
count+=1
else:
for l in range(k+1, len(letters)):
if letters[i]+letters[j]+letters[k]+letters[l]==a:
count+=1
return count
Run Code Online (Sandbox Code Playgroud)
FWIW,使用itertools比使用递归更容易表达某些类型的组合逻辑.例如,当出现此模式时:
letters = 'ABCDEF'
for i in range(len(letters)):
a = letters[i]
for j in range(i+1, len(letters)):
b = letters[j]
for k in range(j+1, len(letters)):
c = letters[k]
print(a, b, c)
Run Code Online (Sandbox Code Playgroud)
它可以替换为:
from itertools import combinations
letters = 'ABCDEF'
for a, b, c in combinations(letters, 3):
print(a, b, c)
Run Code Online (Sandbox Code Playgroud)
你的问题还有很多,但我想指出组合函数是这个问题中提出的那种逻辑的一个很好的起点.