Leetcode 77. 组合 dfs

题目:

给定两个整数 n 和 k,返回 1 ... 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

注意,【1,2】和【2,1】是一种情况。

代码如下:

class Solution {
public:
    vector<vector<int>> ans;
    int vis[1005];
    void dfs (int num,int loc,int k,int n,vector<int>v)
    {
        if(num==k)
        {
            ans.push_back(v);
            return;
        }
        for (int i=loc;i<=n;i++)
        {
            if(!vis[i])
            {
                vis[i]=1;
                v.push_back(i);
                dfs(num+1,i+1,k,n,v);
                v.pop_back();
                vis[i]=0;
            }
        }
    }
    vector<vector<int>> combine(int n, int k) {
        memset (vis,0,sizeof(vis));
        vector<int>v;
        if(n==0)
            return ans;
        dfs(0,1,k,n,v);
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_41410799/article/details/85798148