原题地址: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)