优美的排列——加深对回溯算法层数的理解

leetcode:优美的排列
假设有从 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。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/beautiful-arrangement

一开始对回溯算法层数和数组对应的第i位没理解透彻,导致解题过程中浪费了一些时间,后来在回溯算法中引入一个值用来记录当前是第几层,就能很好的与数组的第i位对应起来啦。做了这题之后加深了对回溯算法层数的理解,下面是代码:

# include<iostream>
# include<vector>
using namespace std;
class Solution {
public:
    int countArrangement(int N) {
        vector<int> resultSet;
        vector<bool> select(N+1,false);
        resultSet.emplace_back(-1);
        int count=0;
        trackback(resultSet,select,N,count,1);
        return count;
    }
    void trackback(vector<int> &resultSet,vector<bool> &select,int N,int &count,int k)
    {
        if(resultSet.size()-1==N)
        {
            count++;
            return;
        }
        for(int i=1;i<=N;i++)
        {
            if(select[i]) continue;
            resultSet.emplace_back(i);
            select[i]=true;
            if(resultSet[k]%k==0||k%resultSet[k]==0)
            trackback(resultSet,select,N,count,k+1);
            resultSet.pop_back();
            select[i]=false;
        }
    }
};

因为题目中说的是i和第i个元素,所以是记录层数的元素k和数组中第k位元素相比较,搞清楚之后就不会出错了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值