Leetcode_526. 优美的排列

题目链接

这道题用dfs写还是很快的，而且效率也挺高的，主要想记录一下状态压缩+dp的方法。官方解答如下：

状态压缩其实就是用二进制表示可能的状态，用mask表示状态，最多有1<<n种状态，第i位（从0开始）为1表示选取了数字i+1,设num(mask)表示mask二进制表示中1的个数，那么f(mask)就是表示排列中前num个位置选用第i位为1对应数字后满足题目两个求余条件之一的组合数目，那么f(mask)可以由比他二进制中1数目少一个的状态转移过来，这些满足条件的子状态相加就是f(mask)对应的方案数。

具体来说就是取mask二进制表示的第i位，第i位表示数字i+1，如果这一位为1表示数字i+1被选取了，只有满足第i位为1并且num%(i+1)0 || (i+1) %num0成立的话才能从将mask中第i位为0对应的状态转移过来，也就是代码中的f[mask ^ (1 << i)]，这句话就是把mask中第i位取反，不考虑需要满足求余＝0的条件的话，f(0111)可以由f(0011) f(0101) f(0110)转移过来，加上求余等于0的条件，就只能从f(0011) f(0110) 转移过来了，0101过不来是因为第1位也就是数字2不能放在位置3（0111中1的个数）处。
下述代码摘自评论区大佬对官方解答的注释，自己语言描述不到这种程度，所以引用过来记录一下，并表示感谢！

  // 用 mask 的二进制表示选取状态，n 个数字用 n 位表示，第 i 位为 1 代表数字 i+1 已被选取（i从0开始），n 中 1 的个数 m 代表前 m 位已放置
   // 例如：二进制 100110 共三个1，代表排列的前三位已放置数字，三个1分别在二进制第 1、2、5位置上(从右侧开始，从0开始计数）, 所以 2、3、6三个数字被选取，综合起来就是表示：2 3 6 这三个数字被放到了排列的前三位，三个数字完美排列方式未知，通过枚举 mask 进行计算

     class Solution {
         public int countArrangement(int n) {
             // 用来存储中间结果，f[6] = f[000110] = 数字2、3在前两位时的完美排列数量
             int[] f = new int[1 << n];
             f[0] = 1;
             // 通过 mask 进行枚举，最终目的是为了得到二进制 mask = (11..11)n 时，总的完美排列数 
             for (int mask = 1; mask < (1 << n); mask++) {
                 int num = Integer.bitCount(mask);
                 // 遍历 mask 的每一位，仍以 mask = 100110 为例，此 mask 代表 2 3 6三个数字在排列的前三位
                 // 求三个数字 2 3 6 的完美排列方式，则先确定2 3 6哪些数字能放到第三位，然后累加另外两个数字的完美排列数量来获得
                 // 2 3 6，第三位可以为 6，则 f[100110] += f[000110] (2、3在前两位时的完美排列数量)
                 // 2 3 6，第三位可以为 3，则 f[100110] += f[100010] (2、6在前两位时的完美排列数量)
                 for (int i = 0; i < n; i++) {
                     // mask & (1<<i) 用来判断 mask 第 i 位是否为 1，如果为 1，说明第 i+1 个数字被选取
                     // ((num % (i + 1)) == 0 || (i + 1) % num == 0) 判断被选取的数字 i+1 能否放到位置 num 上，
                     // 即：先从被选取的数字中找到能放到最高位 num 的数字，然后将剩余 num-1 个数字的完美排列方式累加到f[mask]中
                     if ((mask & (1 << i)) != 0 && ((num % (i + 1)) == 0 || (i + 1) % num == 0)) {
                         // mask ^ (1 << i) 将 mask 第 i 位设置为 0
                         f[mask] += f[mask ^ (1 << i)];
                     }
                 }
             }
             return f[(1 << n) - 1];
         }
     }

·
·
·
·
·
·
·
·

附上自己写的dfs:

class Solution {
public:
    bool vis[16]={false}; 
    int res=0;
    int N;
    int countArrangement(int n) {
        N=n;
        dfs(n);
        return res;
    }
    void dfs(int depth){
        if(depth==0){
            res++;
            return;
        }
        for(int i=1;i<=N;i++){
            if(!vis[i]){
                if(i%depth==0 || depth%i==0){
                    vis[i]=true;
                    dfs(depth-1);
                    vis[i]=false;
                }
            }
        }
    }
};