[数据结构与算法]力扣526——优美的排列（回溯）

题目（中等）
假设有从 1 到 n 的 n 个整数。用这些整数构造一个数组 perm（下标从 1 开始），只要满足下述条件 之一 ，该数组就是一个 优美的排列 ：

perm[i] 能够被 i 整除
i 能够被 perm[i] 整除
给你一个整数 n ，返回可以构造的 优美排列 的 数量 。

示例 1：
输入：n = 2
输出：2
解释：
第 1 个优美的排列是 [1,2]：
- perm[1] = 1 能被 i = 1 整除
- perm[2] = 2 能被 i = 2 整除
第 2 个优美的排列是 [2,1]:
- perm[1] = 2 能被 i = 1 整除
- i = 2 能被 perm[2] = 1 整除

提示：
1 <= n <= 15

思路
回溯，设一个访问数组，标记某个数字是否已用

代码

class Solution {
private:
    vector<int> vis = vector<int>(20, 0);
    int ans = 0;
public:    
    void dfs(int pos, int n) {
        if(pos > n) ans++;
        for(int i = 1; i <= n; i++) {
            if(vis[i]) continue;
            if(pos % i == 0 || i % pos == 0) {
                vis[i] = 1;
                dfs(pos + 1, n);
                vis[i] = 0;
            }
        }
        return;
    }
    int countArrangement(int n) {
        dfs(1, n);
        return ans;
    }
};