题目链接:526. 优美的排列
回溯
树型结构:
预处理 m a t c h match match 数组(每个位置符合条件的数有哪些):
void getMatch(int n) {
used.resize(n + 1);
match.resize(n + 1);
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
if (i % j == 0 || j % i == 0) {
match[i].push_back(j);
}
}
}
}
(1) 回溯函数参数及返回值:
i
n
d
e
x
index
index 记录遍历到第几个数字, 表示递归的深度。
u
s
e
d
used
used 数组记录排列里哪些元素已经被使用了,一个排列里一个元素只能使用一次。
m
a
t
c
h
match
match 二维数组用来存放不同的集合(每个位置符合条件的数有哪些),每个
i
n
d
e
x
index
index 对应一个集合。
r
e
s
res
res 表示符合条件的集合的个数。
vector<bool> used;
vector<vector<int>> match;
int res;
void backtracking(int n, int index) {
(2) 终止条件:
由树型结构可知,当
i
n
d
e
x
=
=
n
+
1
index == n + 1
index==n+1, 收集结果并返回。
if (index == n + 1) {
res++;
return;
}
(3) 确定单层递归逻辑:
首先要取
i
n
d
e
x
index
index 对应的
m
a
t
c
h
match
match 里面的数字,然后
f
o
r
for
for 循环来处理这个集合。
一个排列里的一个元素只能使用一次,所以当该元素已经被使用,直接
c
o
n
t
i
n
u
e
continue
continue。
注意递归参数要
i
n
d
e
x
+
1
index + 1
index+1, 因为下一层要处理下一个数字了。
for (auto& x : match[index]) {
if (used[x] == true) {
continue;
}
used[x] = true;
backtracking(n, index + 1);
used[x] = false;
}
代码如下:
class Solution {
private:
vector<bool> used;
vector<vector<int>> match;
int res;
void backtracking(int n, int index) {
if (index == n + 1) {
res++;
return;
}
for (auto& x : match[index]) {
if (used[x] == true) {
continue;
}
used[x] = true;
backtracking(n, index + 1);
used[x] = false;
}
}
void getMatch(int n) {
used.resize(n + 1);
match.resize(n + 1);
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
if (i % j == 0 || j % i == 0) {
match[i].push_back(j);
}
}
}
}
public:
int countArrangement(int n) {
getMatch(n);
backtracking(n, 1);
return res;
}
};