lintcode刷题（python）--排列组合
lintcode中有许多排列组合的题目，这些题目如果要自己写代码的话要用到递归思想。。
和数组的排列一样，python中也内置了排列组合的库函数，在刷题过程中，直接用这些库函数真是爽的很。
排列的函数：
import itertools
print(list(itertools.permutations([1,2,3,4],3)))上面的代码直接打印出[1，2，3，4]中选3个数的全排列，如果第二个参数不填，则默认为数组的长度。
组合的函数：
import itertools
print(list(itertools.combinations([1,2,3,4],3)))和上面的情况类似。
以lintcode的具体题目为例
全排列
给定一个数字列表，返回其所有可能的排列。
注意事项
你可以假设没有重复数字。
import itertools
class Solution:
"""
@param nums: A list of Integers.
@return: A list of permutations.
"""
def permute(self, nums):
# write your code here
result=[]
permutation_result=list(itertools.permutations(nums))
for i in permutation_result:
result.append(list(i))
return result如果使用非递归的形式来写，就要这样：
class Solution:
"""
@param nums: A list of Integers.
@return: A list of permutations.
"""
def permute(self, nums):
if nums is None:
return []
if nums == []:
return [[]]
nums = sorted(nums)
permutation = []
stack = [-1]
permutations = []
while len(stack):
index = stack.pop()
index += 1
while index < len(nums):
if nums[index] not in permutation:
break
index += 1
else:
if len(permutation):
permutation.pop()
continue
stack.append(index)
stack.append(-1)
permutation.append(nums[index])
if len(permutation) == len(nums):
permutations.append(list(permutation))
return permutations
第k个排列
给定 n 和 k，求123..n组成的排列中的第 k 个排列。
注意事项
1 ≤ n ≤ 9
样例
对于 n = 3, 所有的排列如下：
123
132
213
231
312
321
如果 k = 4, 第4个排列为，231.
import itertools
class Solution:
"""
@param n: n
@param k: the k-th permutation
@return: a string, the k-th permutation
"""
def getPermutation(self, n, k):
result = list(itertools.permutations(range(1,n+1)))[k-1]
string=""
for i in result:
string +=str(i)
return string
k数和 II
给定n个不同的正整数，整数k（1<= k <= n）以及一个目标数字。
在这n个数里面找出K个数，使得这K个数的和等于目标数字，你需要找出所有满足要求的方案。
样例
解析：这道题和上面两道题思路都是一样的，只要用一个变量保存当前的最接近值（贪心算法的思想），最后返回这个最接近值即可。
给出[1,2,3,4]，k=2， target=5，返回 [[1,4],[2,3]]
从n个数找出k个数，是组合问题，得到组合后，一个一个和是否为目标数字，即得到结果。
import itertools
class Solution:
"""
@param A: An integer array.
@param k: A positive integer (k <= length(A))
@param target: Integer
@return a list of lists of integer
"""
def kSumII(self, A, k, target):
# write your code here
result=list(itertools.combinations(A,k))
lists=[]
for i in result:
if sum(i)==target:
lists.append(list(i))
return (lists)三数之和
给出一个有n个整数的数组S，在S中找到三个整数a, b, c，找到所有使得a + b + c = 0的三元组。
注意事项
在三元组(a, b, c)，要求a <= b <= c。
结果不能包含重复的三元组。
样例
如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是：
(-1, 0, 1)
(-1, -1, 2)
import itertools
class Solution:
"""
@param numbersbers : Give an array numbersbers of n integer
@return : Find all unique triplets in the array which gives the sum of zero.
"""
def threeSum(self, numbers):
"""
首先要对numbers进行清洗，如果numbers中有的数出现了超过两次，则让其最多出现两次。
例如numbers由100个1和100个-1组成。清洗后的数组的结果和未清洗的结果都是相同的。
但0就不一样了，因为3个0的和也是0，所以0最多可以出现3次
清洗完后，对numbers进行组合，判断组合是否和为0。
注意，返回的结果要按数字大小排序，而且不能有重复的结果。
"""
CleanNumbers=[]
setNumbers=set(numbers)
for i in setNumbers:
if numbers.count(i)>=2:
CleanNumbers.append(i)
CleanNumbers.append(i)
else:
CleanNumbers.append(i)
if numbers.count(0)>=3:
CleanNumbers.append(0)
result=[]
combinations=list(itertools.combinations(CleanNumbers,3))
for i in combinations:
if sum(i)==0 and sorted(list(i)) not in result:
i=list(i)
i.sort()
result.append(i)
return sorted(result)
四数之和
给一个包含n个数的整数数组S，在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。
注意事项
四元组(a, b, c, d)中，需要满足a <= b <= c <= d
答案中不可以包含重复的四元组。
样例
例如，对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为：
(-1, 0, 0, 1)
(-2, -1, 1, 2)
(-2, 0, 0, 2)
这道题和上面的三数之和是一模一样的思路，代码如下：
import itertools
class Solution:
"""
@param numbersbers : Give an array numbersbersbers of n integer
@param target : you need to find four elements that's sum of target
@return : Find all unique quadruplets in the array which gives the sum of
zero.
"""
def fourSum(self, numbers, target):
# write your code here
setNumbers=set(numbers)
cleanedNumbers=[]
for i in setNumbers:
if numbers.count(i)>=3:
cleanedNumbers.append(i)
cleanedNumbers.append(i)
cleanedNumbers.append(i)
elif numbers.count(i)==2:
cleanedNumbers.append(i)
cleanedNumbers.append(i)
else:
cleanedNumbers.append(i)
if numbers.count(0)>=4:
cleanedNumbers.append(0)
result = []
combinations = list(itertools.combinations(cleanedNumbers, 4))
for i in combinations:
if sum(i) == target and sorted(list(i)) not in result:
result.append(sorted(list(i)))
return sorted(result)
最接近的三数之和
给一个包含 n 个整数的数组 S, 找到和与给定整数 target 最接近的三元组，返回这三个数的和。
注意事项
只需要返回三元组之和，无需返回三元组本身
样例
例如 S = [-1, 2, 1, -4] and target = 1. 和最接近 1 的三元组是 -1 + 2 + 1 = 2.
import itertools
class Solution:
"""
@param numbers: Give an array numbers of n integer
@param target : An integer
@return : return the sum of the three integers, the sum closest target.
"""
def threeSumClosest(self, numbers, target):
# write your code here
cleanedNumbers=[]
setNumbers=set(numbers)
for i in setNumbers:
if numbers.count(i)>=3:
cleanedNumbers.append(i)
cleanedNumbers.append(i)
cleanedNumbers.append(i)
elif numbers.count(i)==2:
cleanedNumbers.append(i)
cleanedNumbers.append(i)
else:
cleanedNumbers.append(i)
combinations = list(itertools.combinations(cleanedNumbers,3))
minNumber=sum(combinations[0])
for i in combinations:
if abs(sum(i)-target)带重复元素的排列
给出一个具有重复数字的列表，找出列表所有不同的排列。
样例
给出列表 [1,2,2]，不同的排列有：
[
[1,2,2],
[2,1,2],
[2,2,1]
]import itertools
class Solution:
"""
@param nums: A list of integers.
@return: A list of unique permutations.
"""
def permuteUnique(self, nums):
# write your code here
result=[]
permutations=list(itertools.permutations(nums))
for i in permutations:
if list(i) not in result:
result.append(list(i))
return result