permutations (dfs递归)

Given a collection of distinct integers, return all possible permutations.

Example:

Input: [1,2,3]
Output:
[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]
class Solution {
public:
    vector<vector<int>> permute(vector<int>& nums) {
        /*  1 2 3 4 5
         *  利用dfs 遍历到的长度 等于数组长度时记录当前结果
         *   关键在于当前结果怎么产生
         *   一种做法是: 交换 后进行dfs
         *   即 每次进行所有可能的交换
         * */
        vector<vector<int>> ret;

        dfs(nums, nums.size(), 0, ret);
        return ret;
    }
    void dfs(vector<int> &nums, int size, int index, vector<vector<int>> &ret){
        // index == size 代表已经搜索到一条完整的路径
        if(index == size)   ret.push_back(nums);
        else{
            for(int i=index;i<size;i++){
                swap(nums[i], nums[index]);
                dfs(nums, size, index+1, ret);
                swap(nums[i], nums[index]); // 恢复
            }
        }
    }
};

猜你喜欢

转载自blog.csdn.net/futangxiang4793/article/details/88650252
今日推荐