算法学习之排列问题

Jackey C/C++ 1,465 次浏览 , 没有评论

题目要求:

给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

 

示例 1:

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

输入:nums = [0,1]
输出:[[0,1],[1,0]]
示例 3:

输入:nums = [1]
输出:[[1]]
提示:

1 <= nums.length <= 6
-10 <= nums[i] <= 10
nums 中的所有整数 互不相同

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/permutations

解题代码:

class Solution {
private:
    vector<vector<int>> res;
    vector<bool> used;

    // p中保存了一个有index个元素的排列
    // 向这个排列的末尾添加第index+1个元素,获得一个有index+1个元素的排列
    void generatePermutation(const vector<int>& nums, int index, vector<int>& p) {
        if (index == nums.size()){
            res.push_back(p);
            return;
        }

        for (int i = 0; i < nums.size(); ++i) {
            if (!used[i]) {
                // 将nums[i]添加在p中
                p.push_back(nums[i]);
                used[i] = true;
                generatePermutation(nums, index + 1, p);
                p.pop_back();
                used[i] = false;
            }
        }
        return;
    }
public:
    vector<vector<int>> permute(vector<int>& nums) {
        res.clear();
        if (nums.size() == 0)
            return res;

        used = vector<bool>(nums.size(), false);
        vector<int> p;
        generatePermutation(nums, 0, p);
        return res;
    }
};

 

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注

Go