Question

我有这个字符串："AAABBB"和这个字符串"--"。

如何在递归中找到合并字符串"--AAABBB"的所有排列？

但"AAABBB"必须遵守她的命令。例如：

--AAABBB 
-A-AABBB 
-AA-ABBB 
. 
.. 
. 
.AAABBB--

Answer 1

这是一个递归的生成器实现：

def comb(first_str, second_str):
    if not first_str:
        yield second_str
        return
    if not second_str:
        yield first_str
        return

    for result in comb(first_str[1:], second_str):
        yield first_str[0] + result
    for result in comb(first_str, second_str[1:]):
        yield second_str[0] + result

用字符串输出：

>>> for result in comb("--", "AAABBB"):
    print(result)


--AAABBB
-A-AABBB
-AA-ABBB
-AAA-BBB
-AAAB-BB
-AAABB-B
-AAABBB-
A--AABBB
A-A-ABBB
A-AA-BBB
A-AAB-BB
A-AABB-B
A-AABBB-
AA--ABBB
AA-A-BBB
AA-AB-BB
AA-ABB-B
AA-ABBB-
AAA--BBB
AAA-B-BB
AAA-BB-B
AAA-BBB-
AAAB--BB
AAAB-B-B
AAAB-BB-
AAABB--B
AAABB-B-
AAABBB--

Answer 2

from itertools import combinations

def dashed_comb(base_str, exp_len):
    for comb in combinations(range(exp_len), exp_len-len(base_str)):
        letters = iter(base_str)
        yield ''.join('-' if x in comb else letters.next()
                               for x in xrange(exp_len))

递归的（在这里提供，因为它比itertools快两倍）：

def recursive_dc(base_str, exp_len):
    if len(base_str) == 0:
        yield '-' * exp_len
    elif len(base_str) == exp_len:
        yield base_str
    else:
        for x in recursive_dc(base_str, exp_len-1):
            yield '-' + x
        for x in recursive_dc(base_str[1:], exp_len-1):
            yield base_str[0] + x

示例输出：

>>> for dc in dashed_comb('AAB', 5):
...     print dc
...
...
--AAB
-A-AB
-AA-B
-AAB-
A--AB
A-A-B
A-AB-
AA--B
AA-B-
AAB--

如何找到字符串的排列？蟒蛇

2 个答案: