526. 优美的排列

优美的排列

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

1. 第 i 位的数字能被 i 整除
2. 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 整除

思路+代码+注释：

public int count=0;
    public int countArrangement(int N) {
        /*
        思路：构造一个N位的数组，可以将求解拆分成N个阶段，每个阶段寻找一位数字，如果该数字能被位数整除或者位数能被该位数字整除，那么该数符合要求寻找下一位数字，
        否则回溯选择另一个数字，当N个阶段都寻找完后证明找到一个符合要求的数组count+1并返回
         */
        int[] nums=new int[N];
        for (int i = 1; i <= N; i++) {
            nums[i-1]=i;
        }
        int[] visited=new int[N];
        countArrangement(nums,N,1,visited);
        return count;
    }

    private void countArrangement(int[] nums,int N,int wei,int[] visited)
    {
        if (wei==N+1)
        {

            count++;

            return;
        }
        for (int i = 0; i < N; i++) {
            if (visited[i]==0)
            {
                int num=nums[i];
                if (num%wei==0 || wei%num==0)
                {
                    visited[i]=1;
                    countArrangement(nums,N,wei+1,visited);
                    visited[i]=0;
                }
            }
        }
    }