취미처럼

[백준] 15654번 N과 M (5) 본문

Algorithm/백준

[백준] 15654번 N과 M (5)

sirius 2021. 3. 3. 09:50
https://www.acmicpc.net/problem/15654
 

15654번: N과 M (5)

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

www.acmicpc.net

 

자기자신은 방문 처리하고, 미리 저장한 arr 에서 해당 인덱스의 값을 ans 배열에 저장하여 출력

 

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

public class Main {

    public static int N, M;
    public static int[] ans;
    public static int[] arr;
    public static boolean[] visit;
    public static StringBuffer sb = new StringBuffer();

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

        N = Integer.parseInt(st.nextToken()); // 자연수 개수
        M = Integer.parseInt(st.nextToken()); // 자연수
        ans = new int[M];
        arr = new int[N];
        visit = new boolean[N];

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(arr);
        dfs(0);
        System.out.println(sb);

    }

    private static void dfs(int depth) {
        if(depth == M) {
            for(int val : ans) {
            	sb.append(val).append(" ");
            }
            sb.append("\n");
            return;
        }

        for(int i = 0 ; i < N; i ++) {
            if(!visit[i]) {
                visit[i] = true;
                ans[depth] = arr[i];
                dfs(depth + 1);
                visit[i] = false;
            }
        }

    }
}

'Algorithm > 백준' 카테고리의 다른 글

[백준] 15656번 N과 M (7)  (0) 2021.03.03
[백준] 15655번 N과 M (6)  (0) 2021.03.03
[백준] 15652번 N과 M (4)  (0) 2021.03.02
[백준] 15651번 N과 M (3)  (0) 2021.03.02
[백준] 15650번 N과 M (2)  (0) 2021.03.02
Comments