python实现排列组合

全排列

1.递归实现全排列

#递归实现全排列
import copy
def permutation_recu(nums,position,end):
    if position==end:
        res.append(copy.deepcopy(nums))
        return
    for i in range(position,end):
        if nums[i] in nums[position:i]:
            continue
        nums[position],nums[i]=nums[i],nums[position]
        permutation_recu(nums,position+1,end)
        nums[position],nums[i]=nums[i],nums[position]
arr=['a','a','b','c']
res=[]
permutation_recu(arr,0,len(arr))
print(res)

2.深度优先搜索实现全排列

#深度优先搜索实现全排列
def permutations_dfs(arr):
	res=[]
	arr.sort()
	def helper(arr,temp):
		if not arr:
			res.append(temp)
		for i in range(len(arr)):
			if i>0 and arr[i]==arr[i-1]:
				continue
			helper(arr[:i]+arr[i+1:],temp+[arr[i]])
	helper(arr,[])
	return res
arr=['a','a','b','c']
print(permutations_dfs(arr))

3.库函数实现全排列

#库函数实现全排列
import itertools
arr=['a','a','b','c']
res=[]
for item in itertools.permutations(arr):
    if list(item) not in res:
        res.append(list(item))
print(res)

组合

1.库函数实现单列表组合

#库函数实现组合
import itertools
arr=['a','a','b','c']
res=list(itertools.combinations(arr,2))  #2为所选组合数
print(res)

2.库函数实现多列表组合

#库函数实现多个列表的组合
import itertools
lst1=['1','2','3']
lst2=['a','b','c','d']
lst3=['x','y','z']
res=list(itertools.product(lst1,lst2,lst3))
print(res)

3.回溯法实现多个列表组合

#回溯法求多个列表组合
#主函数
def combination(arr):
    res=[]
    tmp=[]
    dfs(res,tmp,arr,0)
    print(res)

#深度优先遍历
def dfs(res,tmp,arr,i):
    #所有的列表都组合到了
    if i==len(arr):
        res.append(tmp.copy())
    else:
        for item in arr[i]:
            tmp.append(item)
            dfs(res,tmp,arr,i+1)
            tmp.pop()
lst1=['1','2','3']
lst2=['a','b','c','d']
lst3=['x','y','z']
combination([lst1,lst2,lst3])

————————————————
积硅步，以至千里