leetCode 46. Permutations 回溯問題 | Medium

46. Permutations(全排列問題--回溯問題經典)

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

For example,
[1,2,3] have the following permutations:

[
  [1,2,3],
  [1,3,2],
  [2,1,3],
  [2,3,1],
  [3,1,2],
  [3,2,1]
]

題目大意:求一個序列的全排列。

思路:做排列組合時,模擬該過程得到思路。首先選擇集合中的某一個元素作爲第一位,接着在剩下的元素中選擇一個元素作爲第二位,一直到爲空。
上述過程顯然是個遞歸的過程,聲明幫助函數:Helper(vector<vector<int> >& result, vector<int>& tmp, vector<int>& nums)。tmp用來存儲已經選定的元素,nums用來存儲剩下的元素,result存儲所得的情況。顯然當nums爲空的時候遞歸應該返回。
注意:關鍵在於對nums和tmp的處理,一旦tmp中增加了一個元素,nums就要刪除該元素;每次有一個幫助函數返回時都會進行一次回溯,即nums要插入元素,tmp要刪除元素。

代碼如下:(採用典型的回溯來處理)

class Solution {
public:
    void permuteHelper(vector<vector<int> >& result,vector<int> &tmp,vector<int>& nums)
    {
        if(nums.empty())
        {
            result.push_back(tmp);
            return;
        }
        
        for(int i = 0; i < nums.size(); i++)
        {
            int val = nums[i];
            tmp.push_back(val);
            nums.erase(nums.begin()+i);
            permuteHelper(result,tmp,nums);
            nums.insert(nums.begin()+i,val);
            tmp.pop_back();
        }
    }
    vector<vector<int>> permute(vector<int>& nums) {
        vector<vector<int>> result;
        vector<int> tmp;
        permuteHelper(result,tmp,nums);
        
        return result;
    }
};

2016-08-07 15:11:04

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章