排列组合的计算方法

问题

求1,2,3...n的不同排列方式(n!)

思路

普通的排列问题,在python库中甚至有现成的库可以用来解决,我们这里考虑两种方案,首先是回溯的思路(递归):
基于交换元素的回溯实现较为简单,每一次交换循环位置的元素和首元素,直到循环位置抵达末尾,交换完毕后回溯

当然也有其他的思路,比如使用一个mark数组记录当前去过的位置,然后每次找下一个没去过的位置,这种方法其实也属于回溯,比较容易理解其正确性

其次是非递归的实现,这里我还没有没有找到与递归等同效率的思路(目前时间复杂度要高出On),只能暂时记录下来留待之后改进了

这里主要思想是既然一共有n!个排列,那么n的排列相当于n-1的所有排列*n,得到n-1的所有排列后,其实只需在每种排列上加上1-n就行了

解决

# 递归
def rec(a, l, r):
    if l==r:
        print(*a)
        return
    for i in range(l, r+1):#注意这里下限是l,虽然把自己跟自己交换是一种重复,但这也是需要计算在内的
        a[i], a[l] = a[l], a[i]
        rec(a,l+1,r)
        a[i], a[l] = a[l], a[i]
rec([i+1 for i in range(n)], 0, n-1)
# 非递归
n = 4
t, steps = 1, [[1]]
while t < n:
    cur, t = [], t+1
    for i in range(1, t+1):
        for s in steps:
            cur.append([i, *[(i+x-1) % t+1 for x in s]])
    steps = cur
    print(*steps) #[1, 2, 3, 4] [1, 2, 4, 3] [1, 3, 4, 2] [1, 3, 2, 4] [1, 4, 2, 3] [1, 4, 3, 2] [2, 3, 4, 1] [2, 3, 1, 4]...

问题

求1-n中取出x个数字的方式(C(n,x))

思路

同样考虑两种方式,首先是递归,记录当前位置和递归深度即可,递归深度为x即取数完毕

第二是非递归,这里运用了一点bit magic,大概意思是根据x二进制中最大递增后缀序列求出下一个排列,具体可以参见这篇文章

解决

# 递归
def comb(n,x,cur,start,depth,arr):
    if depth==x:
        arr.append(cur)
        return
    for next in range(start,n-x+depth+1):
        comb(n,x,cur+str(next),next+1,depth+1,arr)
comb(n,x,'',0,0,arr) # 01 02 03 04 12 13 14 23 24 34,这里下标是0开始的,不影响效果
# 非递归
n, x = 5, 3
start, end = (1 << c)-1, (1 << n) - (1 << (n-x)) # 这里start和end就是二进制中x个1在最开始和最末尾的数
v = start
while v <= end:
    # 这里结果所对应的二进制位为1的下标就是具体的组合
    print(bin(v)[2:].rjust(n, '0'))
    # 下面两行是求next permutation的位运算方法,具体解释可以参考引用的两篇资料
    t = (v | (v - 1)) + 1 
    w = t | ((int((t & -t) / (v & -v)) >> 1) - 1)
    v = w

Ref

https://www.geeksforgeeks.org/write-a-c-program-to-print-all-permutations-of-a-given-string/
http://blog.gaurav.im/2016/12/18/next-binary-permutation-bitwise-hackery/
https://graphics.stanford.edu/~seander/bithacks.html#NextBitPermutation

Tips

  • x & -x是求x的二进制中最右边的1所对应位置的二进制数,比如x=01100100,x&-x=00000100,x为奇数这个值就一定是1
  • x | x-1是将x的二进制中所有后继0置1,比如x=01100100,x&-x=01100111,同样x为奇数这个值就是本身
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。