一道面试题--------[1,n]区间选k个数的组合排列

前言

面试的时候需要手撕代码，面试官出了一道题，初看很简单，写起来却发现挺绕的，当时花了30分钟都没绕明白，赶紧让面试官换了一道，第二道就简单一些，10分钟就写完了，还好有惊无险。
其实是leetcode的一道题。这件事告诉了我，还是得有事没事儿，刷一刷leetcode，不说碰不碰到原题，一些算法思想还是得熟悉和巩固。

题目

题目是这样的，输入n和k，输出[1,n]区间，k个数的组合。
Example:

Input: n = 4, k = 2
Output:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

分析：这道题，一看就要用递归，因为无法嵌套循环。
用递归的思想去解决问题时，有两点要注意
1.递归的时候一定要分清楚，边界条件，前进段，返回段。
2.递归先考虑纵向，再考虑横向。
先说一个最常见的思路，比如说n=5,k=3。根据排列组合公式，C=n!/[(n-m)!*m!]=5!/(2!*3!)=10种。
我们在寻找1,2,3,4,5中的3个数，思路是先找1,2,3->1,2,4->1,2,5 ，最后一个数到头了，再找1,3,4->1,3,5，然后是1,4,5，依此类推，1用完了，改变第一个数2,3,4->2,3,5->2,4,5->3,4,5。刚好10种情况。
我们先考虑纵向
1.我们先考虑把1,2,3依次放进去，代码如下，这样我们就实现了，将1,2,3或者2,3,4这种连续的数的存放。

vector<int> out;
vector<vector<int>> res;
void fun(int index,int n,int k)
{
	if(out.size()==k)
	{
		res.push_back(out);
		return;
	}
	out.push_back(index);
	fun(index+1,n,k);
}

纵向我们考虑的是怎么去放3个数进去，我们再考虑横向，我们第一个数可以是1,2,3,4,5。当我们第一个数是4的时候，最多把5推进去，out的大小不会等于3，所以并不会在res留有结果。
我们第一层的分支是1,2,3,4,5实际上，只有1,2,3。4和5满足不了边界条件。
第二层的分支是在第一层的基础上+1，例如第一个数是1，则第二层的选择是2,3,4,5，实际情况只有2,3,4，对于5不满足边界条件。第三层，如果一二层选择的是1,2，则第三层可以选择3,4,5。
因为这个边界条件可以过滤掉不正确的选择，所以我们的横向只要一个index到n的遍历即可。

vector<int> out;
vector<vector<int>> res;
void fun(int index,int n,int k)
{
	if(out.size()==k)
	{
		res.push_back(out);
		return;
	}
	for(int i=index;i<=n;i++)
	{
		out.push_back(i);//当前层数放入out
		fun(i+1,n,k);//进行下一层操作
		res.pop_back();//回溯弹出
	}	
}

完整的代码如下

#include <bits/stdc++.h>
using namespace std;

vector<int> out;
vector<vector<int>> res;
void fun(int index, int n, int k)
{
    if (out.size() == k)
    {
        res.push_back(out);
        return;
    }
    for (int i = index; i <= n; i++)
    {
        out.push_back(i);
        fun(i + 1, n, k);
        out.pop_back();
    }
}

int main()
{
    int n, k;
    while (cin >> n >> k)
    {
        fun(1, n, k);

        for (auto c : res)
        {
            cout << '[';
            for (vector<int>::iterator it=c.begin();it!=c.end();it++)
            {
                if(it==c.end()-1)
                {
                         cout << *it ;
                }
                else{
                        cout << *it << ',';
                }
               
            }
            cout << ']'<<endl;
        }
    }

    return 0;
}