【LeetCode】526. 优美的排列 解题报告 (python)

原题地址：https://leetcode-cn.com/problems/beautiful-arrangement/submissions/

题目描述：

假设有从 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。

 

解题方案：

采用递归方式的回溯法，首先进行全排列，再进行筛选：

class Solution:
    def countArrangement(self, N: int) -> int:
        res = [False] * N
        ans = []

        def dps(ltmp):
            if ltmp[-1] % len(ltmp) != 0 and len(ltmp) % ltmp[-1] != 0:
                return
            if len(ltmp) == N:
                ans.append(ltmp[:])
            else:
                for i in range(N):
                    if res[i] == True:
                        continue
                    res[i] = True
                    dps(ltmp + [i + 1])
                    res[i] = False
        dps([1])
        return len(ans)

其他方法：

class Solution:
    def countArrangement(self, N: int) -> int:
        def fc(li, n):
            count = 0
            
            if n ==1:
                return 1 
            for l in li:
                if l%n == 0 or n%l == 0:
                    tmp = li[:]
                    tmp.remove(l)
                    count += fc(tmp,n-1)
            return count
        begin = range(1,N+1)
        return fc(list(begin),N)