[백준] 15652번 : N과 M (4)

 

문제

자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

• 1부터 N까지 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.
• 고른 수열은 비내림차순이어야 한다.
  • 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

 

입력

 

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

 

출력

 

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다.

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

풀이 방법

 

중복으로 값이 사용이 되므로 중복순열이며, 동시에 비내림차순(오름차순)의 형태를 가지고 있다.

중복순열이므로 boolean을 통한 값 비교처리가 없으며, 오름차순의 형태를 위해서는

순환호출을 할 때 뽑는 개수 (depth)의 증가와 함께, 현재 탐색하고 있는 배열의 인덱스를 같이 넣어줘야 한다.

 

작성 코드

 

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39

import java.util.*; import java.io.*;   public class Main {       static int N,M;     static int output[];     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());         output=new int[N];           permutation(0,0);         System.out.println(sb);       }       public static void permutation(int idx, int depth){           if(depth==M){             for(int i=0;i<M;i++){                 sb.append(output[i]+" ");             }             sb.append("\n");             return;         }           for(int i=idx;i<N;i++){             output[depth]=i+1;             permutation(i,depth+1);         }     } }