题解 | #牛棚分组#

题目考察的知识点是：

递归和回溯算法。

题目解答方法的文字分析：

核心思想是使用回溯算法，在生成符合要求的牛的组合方案时，每次都选择添加当前牛或不添加当前牛，并判断当前已经添加的牛数量是否满足要求。如果满足要求，则保存当前的组合方案，然后进行回溯（即删除最近添加的牛），继续向下遍历。当所有可能的组合方案都被遍历完后，返回包含所有符合要求的组合方案的列表。。

本题解析所用的编程语言：

java语言。

完整且正确的编程代码：

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param n int整型
     * @param k int整型
     * @return int整型二维数组
     */
    public static ArrayList<ArrayList<Integer>> lists = new
    ArrayList<ArrayList<Integer>>();
    public int[][] combine (int n, int k) {
        // write code here
        search(1, k, n, new ArrayList<>());
        int[][] arr = new int[lists.size()][k];
        for (int i = 0; i < lists.size(); i++) {
            for (int j = 0; j < lists.get(i).size(); j++) {
                arr[i][j] = lists.get(i).get(j);
            }
        }
        return arr;
    }

    public void search(int cur, int capacity, int n, ArrayList<Integer> list) {
        if (list.size() == capacity) {
            lists.add(new ArrayList<>(list));
            return;
        }
        if (cur > n) {
            return;
        }
        list.add(cur);
        search(cur + 1, capacity, n, list);
        list.remove(list.size() - 1);
        search(cur + 1, capacity, n, list);
    }

}