leetcode 526. 优美的排列 -二进制数-动态规划-全排列

题目来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems
特别鸣谢：来自夸夸群的 醉笑陪公看落花@知乎，王不懂不懂@知乎，QFIUNE@csdn
感谢醉笑陪公看落花@知乎 倾囊相授，感谢小伙伴们督促学习，一起进步

假设有从 1 到 N 的 N 个整数，如果从这 N 个数字中成功构造出一个数组，使得数组的第 i 位 (1 <= i <= N) 满足如下两个条件中的一个，我们就称这个数组为一个优美的排列。条件：

第 i 位的数字能被 i 整除
i 能被第 i 位上的数字整除
现在给定一个整数 N，请问可以构造多少个优美的排列？

示例1:

输入: 2
输出: 2
解释:

第 1 个优美的排列是 [1, 2]:
第 1 个位置（i=1）上的数字是1，1能被 i（i=1）整除
第 2 个位置（i=2）上的数字是2，2能被 i（i=2）整除

第 2 个优美的排列是 [2, 1]:
第 1 个位置（i=1）上的数字是2，2能被 i（i=1）整除
第 2 个位置（i=2）上的数字是1，i（i=2）能被 1 整除
说明:

N 是一个正整数，并且不会超过15。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/beautiful-arrangement

检查交换实现的全排列的第i个元素是否满足条件，DFS- 通过

参考
leetcode 46 和47 . 全排列,两种生成树的方案来解决全排列问题

这里用方法2来实现全排列，每一次判断红色圈里的数字是否满足条件，如果满足条件，树才继续往下面生长，从而实现了剪枝

class Solution1:
    def countArrangement(self, n: int) -> int:
        nums = list(range(1, n + 1))
        ans =[0]
        def DFS(i):
            if i>0 and not (nums[i-1]%(i)==0 or i% nums[i-1]==0):return
            if i ==len(nums):
                print(nums)
                ans[0] +=1
                return
            for j in range(i,len(nums)):
                nums[i],nums[j] = nums[j],nums[i]
                DFS(i+1)
                nums[i],nums[j] = nums[j],nums[i]

        DFS(0)
        return ans[0]

基于交换的全排列，剪枝优化 - 通过

第i个位置的时候，可能的选择不是全体数字，因此只将i位和能放在i位的数字进行交换，实现剪枝

由于数字是不重复的，ly 变量用来存储某个位置可以使用的数字set()
用set的原因是后面判断是否是可用数字的时候，查找的时间复杂度 为 O(1)

class Solution2:
    def countArrangement(self, n: int) -> int:
        nums = list(range(1, n + 1))
        ly = [set() for _ in range(n+1)]
        for i in range(1,n+1): # 位置
            for j in range(1,n+1):  # 数值
                if (i % j == 0 or j % i == 0):
                    ly[i].add(j)
        ans =[0]
        def DFS(i):
            if i ==len(nums):
                ans[0] +=1
                return
            for j in range(i,len(nums)):
                if nums[j] not in ly[i+1]:continue
                nums[i],nums[j] = nums[j],nums[i]
                DFS(i+1)
                nums[i],nums[j] = nums[j],nums[i]
        DFS(0)
        return ans[0]

官方题解，二进制数，动态规划- 通过

dp数组定义：dp[mask] = x: 已知n的情况下，当num(mask) 个数字被使用时，有x种满足条件的情况

初始状态：使用0个数字的时候，排列方案是1个
状态转移函数：已知n的时候，

class Solution3:
    def countArrangement(self, n: int) -> int:
        f = [0] * (1 << n)
        f[0] = 1  # dp 数组初始值
        for mask in range(1, 1 << n):
            num = bin(mask).count("1")
            for i in range(n):
                if mask & (1 << i) and (num % (i + 1) == 0 or (i + 1) % num == 0):
                    f[mask] += f[mask ^ (1 << i)]
        return f[(1 << n) - 1]

运行效果