[Java] 6603 로또

문제

문제
독일 로또는 {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가 주어졌을 때, 수를 고르는 모든 방법을 구하는 프로그램을 작성하시오.

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

입력의 마지막 줄에는 0이 하나 주어진다. 

출력
각 테스트 케이스마다 수를 고르는 모든 방법을 출력한다.
이때, 사전 순으로 출력한다.

각 테스트 케이스 사이에는 빈 줄을 하나 출력한다.

문제 풀이

순열로 풀 수도 있고 DFS를 사용할 수 도 있다. 이번에는 DFS로 풀었다.

출력되는 결과들은 오름차순으로 정렬되어 있어야하며, 총 6자리로 이루어진 숫자여야한다.

주어지는 숫자는 0 : 주어지는 숫자 갯수, 1 ~ N : 오름차순으로 주어진 숫자들

주어진 N개 중에 6개를 뽑는 것이기 때문에 반복문은 N - 6 + 1이된다.

예를 들어 7 1 2 3 4 5 6 7 이 주어졌다면

배열의 크기는 8이 되고 N은 7이다.

이때 인덱스는 1부터 7 - 6 + 1 = 2까지 반복문을 돌리면된다.

그 이상인 3부터는 6개를 뽑아낼 수 없기 때문에 의미가 없다.

DFS 파라미터로는 현재까지 뽑아낸 숫자의 깊이 (depth), 현재까지 완성된 숫자 (line), 현재까지 완성된 숫자 중 가장 큰 수(num)이 주어진다.

depth번째 숫자부터 확인하는데,

계산에 한 번도 쓴적이 없는 숫자이면서 num보다 큰 숫자만 골라내어 다음 line에 붙여나간다.

depth이 6이되면 6자리 숫자가 된 것이기 때문에 list에 저장하고 종료한다.

 

계산이 종료된 뒤 list에 들어있는 문자열들을 차례로 출력한다.

 

자바 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;

public class Lotto {
    private static final List<String> list = new ArrayList<>();
    private static String[] nums;
    private static boolean[] visited;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        while(true){
            nums = br.readLine().split(" ");
            visited = new boolean[nums.length];

            if(nums[0].equals("0"))
                return;

            // 6자리를 골라야하기 때문에 범위를 nums - 6으로 잡아준다.
            for(int i = 1; i <= nums.length - 6; i++) {
                dfs(1, nums[i], nums[i]);
            }

            for(String s : list){
                System.out.println(s);
            }

            System.out.println();
            list.clear();
        }
    }

    private static void dfs(int depth, String line, String num){
        if(depth == 6){
            list.add(line);
            return;
        }

        for(int i = depth; i < nums.length; i++){
            if(!visited[i] && (Integer.parseInt(nums[i]) > Integer.parseInt(num))) {
                visited[i] = true;
                dfs(depth + 1, line + " " + nums[i], nums[i]);
                visited[i] = false;
            }
        }
    }
}