题目描述
给出一列互不相同的整数,返回其全排列。
样例
输入:
[1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]
算法
(回溯) $O(n \times n!)$
我们从前往后,一位一位枚举,每次选择一个没有被使用过的数。
选好之后,将该数的状态改成“已被使用”,同时将该数记录在相应位置上,然后递归。
递归返回时,不要忘记将该数的状态改成“未被使用”,并将该数从相应位置上删除。
时间复杂度分析:
- 搜索树中最后一层共 $n!$ 个叶节点,在叶节点处记录方案的计算量是 $O(n)$,所以叶节点处的计算量是 $O(n \times n!)$。
- 搜索树一共有 $n! + \frac {n!}{2!} + \frac{n!}{3!} + … = n!(1 + \frac{1}{2!} + \frac{1}{3!} + … ) \le n!(1 + \frac{1}{2} + \frac{1}{4} + \frac{1}{8} + …) = 2n!$ 个内部节点,在每个内部节点内均会for循环 $n$ 次,因此内部节点的计算量也是 $O(n \times n!)$。 所以总时间复杂度是 $O(n \times n!)$。
C++ 代码
class Solution {
public:
vector<vector<int>> ans;
vector<bool> st;
vector<int> path;
vector<vector<int>> permute(vector<int>& nums) {
for (int i = 0; i < nums.size(); i ++ ) st.push_back(false);
dfs(nums, 0);
return ans;
}
void dfs(vector<int> &nums, int u)
{
if (u == nums.size())
{
ans.push_back(path);
return ;
}
for (int i = 0; i < nums.size(); i ++ )
if (!st[i])
{
st[i] = true;
path.push_back(nums[i]);
dfs(nums, u + 1);
st[i] = false;
path.pop_back();
}
}
};
DFS的讲解视频里说复杂度应该是O(n*n!),所以是不是要改一下这个题解?
是的!
这里忘记加上记录方案的计算量了,每个答案记录方案需要 $O(n)$ 的计算量,所以总时间复杂度是 $O(n \times n!)$
太NB了
为什么每个节点会循环n次呢
请问st是什么的缩写?
state
其实我一直好奇一个问题, 为什么dfs要还原现场?
回溯的精髓啊,在某一条路径(dfs)走到底往回走的时候,要把走过的印记擦除掉,如果不擦除的话,那么整个dfs就只会返回一条路径
get!谢谢大佬解答
从一个点往不同分支走的时候要保证不同分支相互之间没有影响才行,所以从每个分支退回来的时候需要把修改过的状态还原。
学会啦,谢谢y总
当n==3的时候,全排列1,2,3:倒数第二层的树的个数不是(n-1)!啊?
这里确实写错了。现在时间复杂度分析部分已重写。
感谢!
nb呀
谢谢hh
别忘了 STL next_permutation哦
最好自己实现一遍,锻炼一下~