1. 程式人生 > >Leetcode 77.組合

Leetcode 77.組合

emp ava 輸出 combine div ... int 提高 時間

組合

給定兩個整數 nk,返回 1 ... n 中所有可能的 k 個數的組合。

示例:

輸入: n = 4, k = 2

輸出:

[

[2,4],

[3,4],

[2,3],

[1,2],

[1,3],

[1,4],

]

采用回溯算法。這是一道 NP 難問題,時間復雜度沒辦法提高,用一個循環遞歸處理子問題,問題的終止條件是每個組合中的元素個數達到 k 個

 1 import java.util.ArrayList;
 2 import java.util.List;
 3 
 4 public class Solution{
 5     public
List<List<Integer>> combine(int n, int k){ 6 List<List<Integer>> res=new ArrayList<List<Integer>>(); 7 List<Integer> temp=new ArrayList<Integer>(); 8 dfs(res,temp,n,k,1); 9 return res; 10 } 11 12 private void
dfs(List<List<Integer>> res,List<Integer> temp,int n,int k,int m){ 13 if(k==0){ 14 res.add(new ArrayList<Integer>(temp)); 15 return; 16 } 17 for(int i=m;i<=n;i++){ 18 temp.add(i); 19 dfs(res,temp,n,k-1,i+1);
20 temp.remove(temp.size()-1); 21 } 22 } 23 }

Leetcode 77.組合