ALGORITHM

[JAVA] 백준 6603- 로또

연듀 2022. 11. 29. 16:12

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

 

6603번: 로또

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로

www.acmicpc.net

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    public static int[] arr, nums;
    public static boolean[] visited;
    public static int k;
    static StringBuilder sb = new StringBuilder();

    public static void dfs(int cnt, int start) {
        if (cnt == 6) {
            for (int x : arr) {
                sb.append(x + " ");
            }
            sb.append("\n");
            return;
        }

        for (int i = start; i < k; i++) {
            if (visited[i]) continue;

            visited[i] = true;
            arr[cnt] = nums[i];
            dfs(cnt + 1, i + 1);
            visited[i] = false;
        }
    }

    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;

            nums = new int[k];
            arr = new int[6];
            visited = new boolean[k];

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

            dfs(0, 0);
            sb.append("\n");
        }
        System.out.println(sb);
    }
}

 

 

순열을 오름차순으로 출력해야 하므로 start변수를 두어  i를 start부터 시작하게 해 dfs를 호출할때마다 1씩 증가시킨다.

애초에 문제에서 집합 S의 원소가 오름차순으로 주어졌기 때문에 이렇게 하는것이 가능하다.