[JAVA] 백준 (실버2) 6603번 로또

링크

https://www.acmicpc.net/problem/6603

---

문제 설명

정답률 55.665%

독일 로또는 {1, 2, ..., 49}에서 수 6개를 고른다.

로또 번호를 선택하는데 사용되는 가장 유명한 전략은 49가지 수 중 k(k>6)개의 수를 골라 집합 S를 만든 다음 그 수만 가지고 번호를 선택하는 것이다.

예를 들어, k=8, S={1,2,3,5,8,13,21,34}인 경우 이 집합 S에서 수를 고를 수 있는 경우의 수는 총 28가지이다. ([1,2,3,5,8,13], [1,2,3,5,8,21], [1,2,3,5,8,34], [1,2,3,5,13,21], ..., [3,5,8,13,21,34])

집합 S와 k가 주어졌을 때, 수를 고르는 모든 방법을 구하는 프로그램을 작성하시오.

---

입력 예제

7 1 2 3 4 5 6 7
8 1 2 3 5 8 13 21 34
0

---

출력 예제

1 2 3 4 5 6
1 2 3 4 5 7
1 2 3 4 6 7
1 2 3 5 6 7
1 2 4 5 6 7
1 3 4 5 6 7
2 3 4 5 6 7

...
3 5 8 13 21 34

---

풀이

길이가 k인 각 수열의 부분수열의 개수가 6이 되는 경우를 모두 출력하면 된다. 각 원소마다 숫자를 저장해가며, 중복없이 오름차순으로 수를 저장해야 되기 때문에 시작 인덱스를 1씩 증가시켜 가며 재귀를 호출한다. depth가 6이 되면 6개의 번호가 저장된 것이므로 재귀를 중단한다.

전체 코드

//백준
public class Main {

    static int k;
    static int[] S;
    static int[] result = new int[6];
    static StringBuilder sb = new StringBuilder();

    public static void main(String[] args) throws IOException {

        System.setIn(new FileInputStream("src/input.txt"));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        while (true) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            k = Integer.parseInt(st.nextToken());
            if (k == 0) break;

            S = new int[k];
            for (int i = 0; i < k; i++) {
                S[i] = Integer.parseInt(st.nextToken());
            }

            recur(0, 0);
            sb.append("\n");
        }

        System.out.println(sb);
    }

    static void recur(int depth, int start) {
        if (depth == 6) {  //depth가 6이 되면 리턴
            for (int num : result) {
                sb.append(num).append(" ");
            }
            sb.append("\n");
            return;
        }

        //오름차순으로 저장하기 위해 start부터 반복
        for (int i = start; i < k; i++) {
            result[depth] = S[i];  //현재 숫자 저장
            recur(depth + 1, i + 1);
        }
    }
}