[BaekJoon] 6603번 - 로또 [Java][C++]

문제 링크

---

1. 문제 풀이

$1$ 부터 $49$ 사이의 수 중 하나인 $k$ 개의 수들에 대해 $6$ 개를 뽑는 모든 조합을 구하는 문제로 재귀를 활용해서 구하면 된다.

---

2. 코드

1. 풀이 [Java]

import java.io.*;
import java.util.*;

public class Main {

    static StringBuilder sb = new StringBuilder();
    static int k;
    static int[] arr;
    static int[] sel;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

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

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

            dfs(0, 0);

            sb.append("\n");
        }

        System.out.println(sb);
    }

    static void dfs(int idx, int selIdx) {
        if (selIdx == 6) {
            for (int x : sel) {
                sb.append(x).append(" ");
            }
            sb.append("\n");
            return;
        }

        for (int i = idx; i < k; i++) {
            sel[selIdx] = arr[i];
            dfs(i + 1, selIdx + 1);
        }
    }
}

2. 풀이 [C++]

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

int k;
int arr[13];
int sel[6];

void dfs(int idx, int selIdx) {
    if (selIdx == 6) {
        for (int x : sel) {
            cout << x << ' ';
        }
        cout << '\n';
        return;
    }

    for (int i = idx; i < k; i++) {
        sel[selIdx] = arr[i];
        dfs(i + 1, selIdx + 1);
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    while (true) {
        cin >> k;
        if (k == 0) break;

        for (int i = 0; i < k; i++) cin >> arr[i];

        dfs(0, 0);
        cout << '\n';
    }
}

---