gpt4 book ai didi

python - 生成所有具有字符 a、b 或 c 且长度等于 n 且最多有 1 个 b 和 2 个 c 的字符串

转载 作者:太空宇宙 更新时间:2023-11-03 20:47:16 24 4
gpt4 key购买 nike

生成具有以下约束的所有字符串

  • 长度:n
  • 允许使用的字符:a、b、c
  • 最多一个 b
  • 最多两个c

我编写了以下代码

def generate(s,counta,countb,countc,n,result):
if counta+countb+countc==n:
print(counta,countb,countc)
result.append(s)
print(s)
return
if counta+countb+countc<n and counta<=n:
generate(s+"a",counta+1,countb,countc,n,result)
if counta+countb+countc<n and countb<=1:
generate(s+"b",counta+1,countb+1,countc,n,result)
if counta+countb+countc<n and countc<=2:
generate(s+"c",counta,countb,countc+1,n,result)


result=[]
generate("",0,0,0,3,result)
print(result)

我得到以下结果,但我不明白为什么。一些长度小于 n 的字符串将添加到结果中。

['aaa', 'aac', 'ab', 'aca', 'acc', 'ba', 'bc', 'caa', 'cac', 'cb', 'cca', 'ccc']

更新代码:(工作)

def generate(s,counta,countb,countc,n,result):
if counta+countb+countc==n:
# print(counta,countb,countc)
result.append(s)
# print(s)
return
if counta+countb+countc<n and counta<=n:
generate(s+"a",counta+1,countb,countc,n,result)
if counta+countb+countc<n and countb<1:
generate(s+"b",counta,countb+1,countc,n,result)
if counta+countb+countc<n and countc<2:
generate(s+"c",counta,countb,countc+1,n,result)

result=[]
generate("",0,0,0,3,result)
print(result)

输出:

['aaa', 'aab', 'aac', 'aba', 'abc', 'aca', 'acb', 'acc', 'baa', 'bac', 'bca', 'bcc', 'caa', 'cab', 'cac', 'cba', 'cbc', 'cca', 'ccb']

最佳答案

如果您更喜欢更简洁的解决方案,那么下面的代码应该可以完成这项工作。约束逻辑与生成逻辑完全分离。这样,可以独立于生成逻辑添加约束。

from itertools import product

n = 3

for i in map(''.join, product('abc', repeat=n)):
if i.count('b') <= 1 and i.count('c') <= 2:
print(i)

输出:

aaa
aab
aac
aba
abc
aca
acb
acc
baa
bac
bca
bcc
caa
cab
cac
cba
cbc
cca
ccb

关于python - 生成所有具有字符 a、b 或 c 且长度等于 n 且最多有 1 个 b 和 2 个 c 的字符串,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56499149/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com