递归生成重复元素的全排列

首先需要对当前已知的列表或数组(python为列表,c/c++/java为数组)由小到大进行排序,这样相等的元素都会在相邻的位置,在for循环中进行递归,需要声明标记当前的数字是否被使用的vis列表、记录递归过程中的结果的列表rec,将这些变量传递到递归方法中,在for循环中判断当前访问的数字是否与上一个数字相等,如果相等而且上一个数字未被访问的条件下说明递归下去求解的方案是与之前求解过的方案是一样的,所以我们continue跳过这种情况尝试其他的情况,如果当前数字未被访问才继续递归下去,其实仔细一想其中的道理还是挺好理解的,因为相等的元素递归的时候相当于是交换了两个元素的位置但是他们求解的结果是一样的,因为之前已经求解过这个方案了,回溯之后那么前一个数字就会被标记为未访问的状态,所以尝试下一个相等元素往下递归的时候求解的方案是重复的。

代码如下:

from typing import List

count = 0


def permutation(index: int, nums: List[int], rec: List[int], vis: List[int]):
    global count
    if index == len(nums):
        count += 1
        print(rec)
        return
    for i in range(len(nums)):
        # 如果相邻的元素相等而且上一个元素与当前元素相等, 上一个元素未被访问的情况下说明求解下去是存在重复排列的
        if i > 0 and nums[i] == nums[i - 1] and vis[i - 1] == 0: continue
        if vis[i] == 0:
            vis[i] = 1
            rec.append(nums[i])
            permutation(index + 1, nums, rec, vis)
            # 回溯
            vis[i] = 0
            rec.pop()


if __name__ == '__main__':
    nums = [1, 1, 1, 2, 2]
    rec = list()
    vis = [0] * len(nums)
    permutation(0, nums, rec, vis)
    print(count)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值