不使用itertools的所有字符串排列

时间:2019-06-03 10:37:30

标签: python-3.x permutation

可以由给定字符串形成的所有长度的所有可能字符串

输入:

abc

输出:

a b c abc ab ac bc bac bca 
         cb ca ba cab cba acb

我已经尝试过使用它,但是它仅限于字符串abc,我想将其概括化,就像输入“ abcd”一样,我将提供相同的输出。

def perm_main(elems):
    perm=[]
    for c in elems:
        perm.append(c)
    for i in range(len(elems)):
        for j in range(len(elems)):
            if perm[i]!= elems[j]:
                perm.append(perm[i]+elems[j])
    level=[elems[0]]
    for i in range(1,len(elems)):
        nList=[]
        for item in level:
            #print(item)
            nList.append(item+elems[i])
                #print(nList)
            for j in range(len(item)):
                #print(j)
                nList.append(item[0:j]+ elems[i] + item[j:])
                #print(nList)
        level=nList
    perm = perm + nList
    return perm

2 个答案:

答案 0 :(得分:2)

您可能不需要itertools,但是您在documentation中有了解决方案,其中itertools.permutations大致等同于:

def permutations(iterable, r=None):
    # permutations('ABCD', 2) --> AB AC AD BA BC BD CA CB CD DA DB DC
    # permutations(range(3)) --> 012 021 102 120 201 210
    pool = tuple(iterable)
    n = len(pool)
    r = n if r is None else r
    if r > n:
        return
    indices = list(range(n))
    cycles = list(range(n, n-r, -1))
    yield tuple(pool[i] for i in indices[:r])
    while n:
        for i in reversed(range(r)):
            cycles[i] -= 1
            if cycles[i] == 0:
                indices[i:] = indices[i+1:] + indices[i:i+1]
                cycles[i] = n - i
            else:
                j = cycles[i]
                indices[i], indices[-j] = indices[-j], indices[i]
                yield tuple(pool[i] for i in indices[:r])
                break
        else:
            return

或使用product

def permutations(iterable, r=None):
    pool = tuple(iterable)
    n = len(pool)
    r = n if r is None else r
    for indices in product(range(n), repeat=r):
        if len(set(indices)) == r:
            yield tuple(pool[i] for i in indices)

它们都是生成器,因此您需要调用list(permutations(x))来检索实际列表,或者将yields替换为l.append(v),其中l是定义为累加结果的列表而v是产生的值。

对于所有可能的尺寸,请对其进行迭代:

from itertools import chain
check_string = "abcd"
all = list(chain.from_iterable(permutations(check_string , r=x)) for x in range(len(check_string )))

答案 1 :(得分:1)

部分递归解决方案。您只需要使其适用于不同的长度即可:

def permute(pre, str):
    n = len(str)
    if n == 0:
        print(pre)
    else:
        for i in range(0,n):
            permute(pre + str[i], str[0:i] + str[i+1:n])

您可以使用permute('', 'abcd')进行调用,也可以使用另一种方法来简化操作

def permute(str):
    permute('', str)

here借来的答案。

通常,将C / Cpp / Java解决方案的代码转换为Python会更好,因为它们通常从头开始实现事情并且不需要库就可以完成事情。

更新

完整解决方案:

def all_permutations(given_string):
    for i in range(len(given_string)):
        permute('', given_string, i+1)

def permute(prefix, given_string, max_len):
    if len(given_string) <= 0 or len(prefix) >= max_len:
        print(prefix)
    else:
        for i in range(len(given_string)):
            permute(prefix + given_string[i], given_string[:i] + given_string[i+1:], max_len)
>>> all_permutations('abc')
a
b
c
ab
ac
ba
bc
ca
cb
abc
acb
bac
bca
cab
cba