首页
学习
活动
专区
圈层
工具
发布

LeetCode 526. 优美的排列(回溯)

1. 题目

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

  • 第 i 位的数字能被 i 整除
  • i 能被第 i 位上的数字整除

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

代码语言:javascript
复制
示例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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 回溯

代码语言:javascript
复制
class Solution {
public:
    int countArrangement(int N) {
        bool visited[N+1] = {false};//数字放过了吗
        int count = 0;
        bt(1,N,visited,count);
        return count;
    }

    void bt(int i, int &N, bool* visited, int &count)
    {	// i 表示填到第几位了
    	if(i == N+1)
    	{
    		count++;
    		return;
    	}
    	for(int num = 1; num <= N; ++num)
    	{
			if(!visited[num] && (num%i == 0 || i%num == 0))
			{
				visited[num] = true;
				bt(i+1, N, visited, count);
				visited[num] = false;
			}
    	}
    }
};
下一篇
举报
领券