본문 바로가기
알고리즘 PS/DFS

[백준] 15656번 - N과 M(7) (Java)

by 백호루이 2022. 12. 9.
반응형

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

 

15656번: N과 M (7)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열

www.acmicpc.net

 

<문제해석>

1. 입력받은 배열은 DFS 탐색을 하기전에 오름차순으로 먼저 정렬을 한다.

2. 중복방문을 허용하기 때문에 방문배열은 사용하지 않는다.

3. 순서가 바뀌면 다른 조합으로 인식한다.

  - for (i = 0 to N)으로 해서 새로운 탐색을 할 때 처음부터 다시 시작한다.

 

<코드구현 #1>

import java.util.Scanner;
import java.util.Arrays;
import java.lang.StringBuilder;

public class Main {
    static int N, M;
    static int A[];
    static int Ans[];
    static StringBuilder sb = new StringBuilder();

    void inputData(){
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        M = sc.nextInt();
        A = new int[N];
        Ans = new int[M];
        for (int i=0; i<N; i++) {
            A[i] = sc.nextInt();
        }
        sc.close();
    }

    void DFS(int start, int level) {
        // 종료 조건
        if (level == M) {
            for (int x : Ans) {
                sb.append(x).append(' ');
            }
            sb.append('\n');
            return;
        }

        // 재귀호출 조건
        // 중복 허용됨 (방문배열 사용X)
        // 순서 바꾸면 다른 조합 인정(i=0 부터 시작)
        for (int i=start; i<N; i++) {
            Ans[level] = A[i];
            DFS(0, level+1);
        }
    }
    
    void Solve() {
        Arrays.sort(A);
        DFS(0, 0);
        System.out.println(sb);
    }
 
    public static void main(String[] args) {
        Main m = new Main();
        m.inputData(); // 입력함수
        m.Solve();
    }
}
반응형

댓글