LeetCode 526. 优美的排列 【回溯算法】

题目

假设有从 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

回溯

class Solution {
    int ans=0;
    int[] vis;
    List<Integer> tmp=new ArrayList<>();
    public void dfs(List<Integer> tmp,int[] vis){
        if(tmp.size()==(vis.length-1)){
            ans++;
            return;
        }
        for(int i=1;i<vis.length;i++){
            if(vis[i]==0&&(i%(tmp.size()+1)==0||(tmp.size()+1)%i==0)){
                tmp.add(i);
                vis[i]=1;
                dfs(tmp,vis);
                tmp.remove(tmp.size()-1);
                vis[i]=0;
            }
        }
    }
    public int countArrangement(int n) {
        vis=new int[n+1];
        for(int i=1;i<=n;i++){
            tmp.add(i);
            vis[i]=1;
            dfs(tmp,vis);
            tmp.remove(tmp.size()-1);
            vis[i]=0;
        }
        return ans;
    }
}