算法实践：leetcode526 优美的排列

leetcode526 优美的排列

描述

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

1. 第 i 位的数字能被 i 整除
2. i 能被第 i 位上的数字整除

现在给定一个整数 N，请问可以构造多少个优美的排列？

示例:

输入: 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 整除

说明:

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

代码

回溯算法，套用模板

result = []
def backtrack(路径, 选择列表):
    if 满足结束条件:
        result.add(路径)
        return
    for 选择 in 选择列表:
        做选择
        backtrack(路径, 选择列表)
        撤销选择

class Solution {
public:
    int Num;
    int res = 0;
    int countArrangement(int N) {
        Num = N;
        int B[N+1];
        bool visited[N+1];
        memset(B,0,sizeof(B));
        memset(visited,false,sizeof(visited));
        Bactrace(B,visited,1);
        return res;
    }
    void Bactrace(int* B,bool* visited,int cell){
        if(cell>Num){
            res++;
            return;
        }
        for(int i=1;i<=Num;i++){
            if(!visited[i]){
                B[cell] = i;
                visited[i] = true;
                if(B[cell]%cell==0 || cell%B[cell]==0){
                    Bactrace(B,visited,cell+1);
                }
                B[cell] = 0;
                visited[i] = false;
            }
        }
    }
};