Поиск всех возможных перестановок данной строки в python
у меня есть строка. Я хочу создать все перестановки из этой строки, изменив порядок символов в ней. Например, скажем:
x='stack'
Я хочу такой список,
l=['stack','satck','sackt'.......]
В настоящее время я повторяю список приведенных строк, выбирая 2 буквы случайным образом и транспонируя их, чтобы сформировать новую строку, и добавляя ее в набор приведенных l. основываясь на длине строки, я вычисляю количество возможных перестановок и продолжаю итерации до тех пор, пока размер набора не достигнет предела. Должен быть лучший способ сделать это.
17 ответов:
модуль itertools имеет полезный метод перестановок(). документация говорит:
itertools.перестановки (iterable[, r])
вернуться последовательных длина Р перестановки элементов в массиве.
если r не указан или нет, то r по умолчанию имеет длину генерируются итерационные и все возможные полнометражные перестановки.
перестановки создаются в лексикографический порядок сортировки. Итак, если вход iterable сортируется, кортежи перестановок будут производиться в сортированном виде порядок.
вам придется присоединиться к вашим перестановочным буквам в виде строк.
>>> from itertools import permutations >>> perms = [''.join(p) for p in permutations('stack')] >>> perms
['stack', 'stakc', 'stcak', 'stcka', 'stkac', 'stkca', 'satck', 'satkc', 'sactk', 'sackt', 'saktc', 'sakct', 'sctak', 'sctka', 'scatk', 'scakt', 'sckta', 'sckat', 'sktac', 'sktca', 'skatc', 'skact', 'skcta', 'skcat', 'tsack', 'tsakc', 'tscak', 'tscka', 'tskac', 'tskca', 'tasck', 'taskc', 'tacsk', 'tacks', 'taksc', 'takcs', 'tcsak', 'tcska', 'tcask', 'tcaks', 'tcksa', 'tckas', 'tksac', 'tksca', 'tkasc', 'tkacs', 'tkcsa', 'tkcas', 'astck', 'astkc', 'asctk', 'asckt', 'asktc', 'askct', 'atsck', 'atskc', 'atcsk', 'atcks', 'atksc', 'atkcs', 'acstk', 'acskt', 'actsk', 'actks', 'ackst', 'ackts', 'akstc', 'aksc', 'aktcs', 'akcst', 'akcts', 'cstak', 'cstka', 'csatk', 'csakt', 'cskta', 'cskat', 'ctsak', 'ctska', 'ctask', 'ctaks', 'ctksa', 'ctkas', 'castk', 'caskt', 'catsk', 'catks', 'cakst', 'cakts', 'cksta', 'cksat', 'cktsa', 'cktas', 'ckast', 'ckats', 'kstac', 'kstca', 'ksatc', 'ksact', 'kscta', 'kscat', 'ktsac', 'ktsca', 'ktasc', 'ktacs', 'ktcsa', 'ktcas', 'kastc', 'kasct', 'katsc', 'katcs', 'kacst', 'kacts', 'kcsta', 'kcsat', 'kctsa', 'kctas', 'kcast', 'kcats']
если вас беспокоят дубликаты, попробуйте вписать свои данные в структуру с помощью нет дубликатов, как
set
:>>> perms = [''.join(p) for p in permutations('stacks')] >>> len(perms) 720 >>> len(set(perms)) 360
спасибо @pst за указание на то, что это не то, что мы традиционно считаем приведением типа, а скорее вызовом
set()
конструктор.
вы можете получить все N! перестановки без особого кода
def permutations(string, step = 0): # if we've gotten to the end, print the permutation if step == len(string): print "".join(string) # everything to the right of step has not been swapped yet for i in range(step, len(string)): # copy the string (store as array) string_copy = [character for character in string] # swap the current index with the step string_copy[step], string_copy[i] = string_copy[i], string_copy[step] # recurse on the portion of the string that has not been swapped yet (now it's index will begin with step + 1) permutations(string_copy, step + 1)
вот еще один подход, отличный от того, что опубликовали @Adriano и @illerucis. Это лучше выполнения, вы можете проверить это сами, измеряя время:
def removeCharFromStr(str, index): endIndex = index if index == len(str) else index + 1 return str[:index] + str[endIndex:] # 'ab' -> a + 'b', b + 'a' # 'abc' -> a + bc, b + ac, c + ab # a + cb, b + ca, c + ba def perm(str): if len(str) <= 1: return {str} permSet = set() for i, c in enumerate(str): newStr = removeCharFromStr(str, i) retSet = perm(newStr) for elem in retSet: permSet.add(c + elem) return permSet
для произвольной строки "dadffddxcf" потребовалось 1.1336 сек для библиотеки перестановок, 9.125 сек для этой реализации и 16.357 сек для версии @Adriano и @illerucis. Конечно, вы все еще можете оптимизировать его.
вот простая функция для возврата уникальных перестановок:
def permutations(string): if len(string) == 1: return string recursive_perms = [] for c in string: for perm in permutations(string.replace(c,'',1)): revursive_perms.append(c+perm) return set(revursive_perms)
пользователи переполнения стека уже опубликовали некоторые сильные решения, но я хотел показать еще одно решение. Этот я считаю, чтобы быть более интуитивным
идея заключается в том, что для данной строки: мы можем повторить алгоритм (псевдокод):
перестановки = char + перестановки (string-char) для char в строке
надеюсь, что это поможет кому-то!
def permutations(string): """Create all permutations of a string with non-repeating characters """ permutation_list = [] if len(string) == 1: return [string] else: for char in string: [permutation_list.append(char + a) for a in permutations(string.replace(char, ""))] return permutation_list
itertools.permutations
это хорошо, но это не очень хорошо справляется с последовательностями, которые содержат повторяющиеся элементы. Это потому, что внутренне он переставляет индексы последовательности и не обращает внимания на значения элементов последовательности.конечно, можно отфильтровать вывод
itertools.permutations
через набор для устранения дубликатов, но он по-прежнему тратит время на создание этих дубликатов, и если в базовой последовательности есть несколько повторяющихся элементов, будет много дубликатов. Также, использование коллекции для хранения результатов приводит к потере ОЗУ, отрицая преимущество использования итератора в первую очередь.К счастью, есть более эффективные подходы. В приведенном ниже коде используется алгоритм индийского математика 14 века Нараяны Пандиты, который можно найти в статья в Википедии о перестановке. Этот древний алгоритм по-прежнему является одним из самых быстрых известных способов генерации перестановок по порядку, и он довольно надежен, поскольку он правильно обрабатывает перестановки, содержащие повторяющиеся элементы.
def lexico_permute_string(s): ''' Generate all permutations in lexicographic order of string `s` This algorithm, due to Narayana Pandita, is from https://en.wikipedia.org/wiki/Permutation#Generation_in_lexicographic_order To produce the next permutation in lexicographic order of sequence `a` 1. Find the largest index j such that a[j] < a[j + 1]. If no such index exists, the permutation is the last permutation. 2. Find the largest index k greater than j such that a[j] < a[k]. 3. Swap the value of a[j] with that of a[k]. 4. Reverse the sequence from a[j + 1] up to and including the final element a[n]. ''' a = sorted(s) n = len(a) - 1 while True: yield ''.join(a) #1. Find the largest index j such that a[j] < a[j + 1] for j in range(n-1, -1, -1): if a[j] < a[j + 1]: break else: return #2. Find the largest index k greater than j such that a[j] < a[k] v = a[j] for k in range(n, j, -1): if v < a[k]: break #3. Swap the value of a[j] with that of a[k]. a[j], a[k] = a[k], a[j] #4. Reverse the tail of the sequence a[j+1:] = a[j+1:][::-1] for s in lexico_permute_string('data'): print(s)
выход
aadt aatd adat adta atad atda daat data dtaa taad tada tdaa
конечно, если вы хотите собрать полученные строки в список вы можете сделать
list(lexico_permute_string('data'))
или в последних версиях Python:
[*lexico_permute_string('data')]
посмотреть
itertools.combinations
илиitertools.permutations
.
вот немного улучшенная версия illerucis's код для возврата списка всех перестановок строки
s
с различными символами (не обязательно в лексикографическом порядке сортировки), без использования itertools:def get_perms(s, i=0): """ Returns a list of all (len(s) - i)! permutations t of s where t[:i] = s[:i]. """ # To avoid memory allocations for intermediate strings, use a list of chars. if isinstance(s, str): s = list(s) # Base Case: 0! = 1! = 1. # Store the only permutation as an immutable string, not a mutable list. if i >= len(s) - 1: return ["".join(s)] # Inductive Step: (len(s) - i)! = (len(s) - i) * (len(s) - i - 1)! # Swap in each suffix character to be at the beginning of the suffix. perms = get_perms(s, i + 1) for j in range(i + 1, len(s)): s[i], s[j] = s[j], s[i] perms.extend(get_perms(s, i + 1)) s[i], s[j] = s[j], s[i] return perms
почему бы вам просто не сделать:
from itertools import permutations perms = [''.join(p) for p in permutations(['s','t','a','c','k'])] print perms print len(perms) print len(set(perms))
вы не получите дубликат, как вы можете видеть:
['stack', 'stakc', 'stcak', 'stcka', 'stkac', 'stkca', 'satck', 'satkc', 'sactk', 'sackt', 'saktc', 'sakct', 'sctak', 'sctka', 'scatk', 'scakt', 'sckta', 'sckat', 'sktac', 'sktca', 'skatc', 'skact', 'skcta', 'skcat', 'tsack', 'tsakc', 'tscak', 'tscka', 'tskac', 'tskca', 'tasck', 'taskc', 'tacsk', 'tacks', 'taksc', 'takcs', 'tcsak', 'tcska', 'tcask', 'tcaks', 'tcksa', 'tckas', 'tksac', 'tksca', 'tkasc', 'tkacs', 'tkcsa', 'tkcas', 'astck', 'astkc', 'asctk', 'asckt', 'asktc', 'askct', 'atsck', 'atskc', 'atcsk', 'atcks', 'atksc', 'atkcs', 'acstk', 'acskt', 'actsk', 'actks', 'ackst', 'ackts', 'akstc', 'aksct', 'aktsc', 'aktcs', 'akcst', 'akcts', 'cstak', 'cstka', 'csatk', 'csakt', 'cskta', 'cskat', 'ctsak', 'ctska', 'ctask', 'ctaks', 'ctksa', 'ctkas', 'castk', 'caskt', 'catsk', 'catks', 'cakst', 'cakts', 'cksta', 'cksat', 'cktsa', 'cktas', 'ckast', 'ckats', 'kstac', 'kstca', 'ksatc', 'ksact', 'kscta', 'kscat', 'ktsac', 'ktsca', 'ktasc', 'ktacs', 'ktcsa', 'ktcas', 'kastc', 'kasct', 'katsc', 'katcs', 'kacst', 'kacts', 'kcsta', 'kcsat', 'kctsa', 'kctas', 'kcast', 'kcats'] 120 120 [Finished in 0.3s]
def permute(seq): if not seq: yield seq else: for i in range(len(seq)): rest = seq[:i]+seq[i+1:] for x in permute(rest): yield seq[i:i+1]+x print(list(permute('stack')))
вот очень простая версия генератора:
def find_all_permutations(s, curr=[]): if len(s) == 0: yield curr else: for i, c in enumerate(s): for combo in find_all_permutations(s[:i]+s[i+1:], curr + [c]): yield "".join(combo)
Я думаю, что это не так уж и плохо!
def f(s): if len(s) == 2: X = [s, (s[1] + s[0])] return X else: list1 = [] for i in range(0, len(s)): Y = f(s[0:i] + s[i+1: len(s)]) for j in Y: list1.append(s[i] + j) return list1 s = raw_input() z = f(s) print z
from itertools import permutations perms = [''.join(p) for p in permutations('ABC')] perms = [''.join(p) for p in permutations('stack')]
def perm(string): res=[] for j in range(0,len(string)): if(len(string)>1): for i in perm(string[1:]): res.append(string[0]+i) else: return [string]; string=string[1:]+string[0]; return res; l=set(perm("abcde"))
Это один из способов генерации перестановок с рекурсией, вы можете легко понять код, взяв строки 'a','ab' & 'abc' в качестве входных данных.
вы получаете все N! перестановки с этим, без дубликатов.
каждый любит запах своего собственного кода. Просто разделяя тот, который я нахожу самым простым:
def get_permutations(word): if len(word) == 1: yield word for i, letter in enumerate(word): for perm in get_permutations(word[:i] + word[i+1:]): yield letter + perm
эта программа не устраняет дубликатов, но я думаю, что это один из самых эффективных подходов:
s=raw_input("Enter a string: ") print "Permutations :\n",s size=len(s) lis=list(range(0,size)) while(True): k=-1 while(k>-size and lis[k-1]>lis[k]): k-=1 if k>-size: p=sorted(lis[k-1:]) e=p[p.index(lis[k-1])+1] lis.insert(k-1,'A') lis.remove(e) lis[lis.index('A')]=e lis[k:]=sorted(lis[k:]) list2=[] for k in lis: list2.append(s[k]) print "".join(list2) else: break