1. 程式人生 > >46. Permutations - Medium

46. Permutations - Medium

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]
]

 

backtracking

為了防止值重複,用visited陣列標記該元素在每一層遞迴的時候是否被訪問過,注意遞迴到最底層pop元素回到上層的時候,要把此元素的visited狀態改成false

time: O(n!), space: O(n)

class Solution {
    public List<List<Integer>> permute(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        boolean[] visited = new boolean[nums.length];
        backtracking(nums, 0, visited, new ArrayList<>(), res);
        return res;
    }
    
    
private void backtracking(int[] nums, int idx, boolean[] visited, List<Integer> tmp, List<List<Integer>> res) { if(tmp.size() == nums.length) res.add(new ArrayList<>(tmp)); for(int i = 0; i < nums.length; i++) { if(visited[i]) continue; visited[i]
= true; tmp.add(nums[i]); backtracking(nums, i + 1, visited, tmp, res); tmp.remove(tmp.size() - 1); visited[i] = false; } } }