回溯
- 思路:
- 这个问题可以看作有 n 个空格,从左往右选择数字填入空格中,每个数字只能使用一次;
- 定义一个递归函数 dfs(first, output),first 是从左往右的第 first 个位置,output 是当前的排列,则:
- 如果 first = size,则将 output 插入结果数组中;
- 如果 first < size,则从没有选过的数中选一个;
- 可以在原数组中将数分为用过的和没有使用过的两段,迭代至一个确定的排列;
- 在退出返回的时候,要还原到上一次的划分状态;
class Solution {
public:
vector<vector<int>> permute(vector<int>& nums) {
std::vector<std::vector<int>> result;
dfs(result, nums, 0, nums.size());
return result;
}
private:
void dfs(std::vector<std::vector<int>>& result, std::vector<int>& output,
int first, int len) {
if (first == len) {
result.emplace_back(output);
return;
}
for (int i = first; i < len; ++i) {
std::swap(output[i], output[first]);
dfs(result, output, first + 1, len);
std::swap(output[i], output[first]);
}
}
};