ALGORITHM

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

연듀 2022. 9. 1. 10:02

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

 

15654번: N과 M (5)

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

www.acmicpc.net

 

 

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

public class Main{
    public static int n, m;
    public static int[] arr, nums;
   public static boolean[] visit; // 방문 여부
    public static StringBuilder sb = new StringBuilder();

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

        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());

        arr = new int[m];
        visit = new boolean[n+1];

        nums=new int[n+1];

        st= new StringTokenizer(br.readLine());
        for(int i=0; i<n; i++){
            nums[i]=Integer.parseInt(st.nextToken());
        }
        Arrays.sort(nums);

        dfs(0);
        System.out.println(sb);
    }

    public static void dfs(int cnt){

        if(cnt == m){
            for(int val:arr){
                sb.append(val).append(" ");
            }
            sb.append("\n");
            return;
        }

        for(int i=1; i<=n; i++){
            if(!visit[i]){
                visit[i] = true;
                arr[cnt] = nums[i];
                dfs(cnt+1);
                visit[i]=false;
            }
        }

    }
}

 

 

N개의 수를 배열로 입력받고,

사전 순으로 증가하는 순서로 출력해야하므로 배열을 오름차순 정렬시켰다.

방문처리를 해 방문하지 않은 값이 나오도록 한다.