优美的排列--回溯解决

Posted 2021-09-06 C_YCBX Py_YYDS

文章目录

• • 题目
  • 提交代码

题目

提交代码

class Solution {
public:
    int countArrangement(int n) {
        unsigned int check = 0;
        //两层循环预处理，得到每个位置保持优美的可取数字(i为数字所在位置序号，j为所选数字)
        vector<vector<int> >select(n);
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i%j==0||j%i==0){
                    select[i-1].emplace_back(j);
                }
            }
        }
        int cnt = 0;
        function<void(int)>backtrack = [&](int pos){
            if(pos>=n){
                cnt++;
                return;
            }
            for(auto&&t:select[pos]){
                if(check&(1<<t))
                    continue;
                check = check|(1<<t);
                backtrack(pos+1);
                check = ~((~check)|(1<<t));
            }
        };
        backtrack(0);
        return cnt;
    }
};