【Leetcode】46. 全排列

题目描述

给定一个没有重复数字的序列，返回其所有可能的全排列。

示例:
输入: [1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]

算法设计

深搜。通过递归实现找到以每个数字开头的排列，用一个tag数组作为使用位来标记元素是否已经放入当前排列中。

C++代码

class Solution {
public:
    void func(vector<int> nums, set<vector<int> > &res, int index, vector<int> tmp, int tag[]){
        if(tmp.size() == nums.size()){
            res.insert(tmp);
            return;
        }
        for(int i=index;i<nums.size();i++){
            if(tag[i] == 0) {
                tmp.push_back(nums[i]);
                tag[i] = 1;
            }
            else continue;
            func(nums, res, 0, tmp, tag);
            tmp.pop_back();
            tag[i] = 0;
        }
    }
    
    vector<vector<int>> permute(vector<int>& nums) {
        set<vector<int> > ress;
        vector<int> tmp;
        vector<vector<int> > res;
        if(nums.size() < 2){
            res.push_back(nums);
            return res;
        }
        int tag[nums.size() + 1];
        memset(tag, 0, sizeof(tag));
        func(nums, ress, 0, tmp, tag);
        set<vector<int> >::iterator it = ress.begin();
        for(it;it!=ress.end();it++){
            res.push_back(*it);
        }
        return res;
    }
};