백준 15664번: N과 M (10) - javascript

백준 15664번: N과 M (10)

문제 설명

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

• N개의 자연수 중에서 M개를 고른 수열
• 고른 수열은 비내림차순이어야 한다.
  • 길이가 K인 수열 A가 A1 ≤ A2 ≤ … ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

입력

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

둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

출력

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

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

접근

왠지 풀어 본 것 같은 느낌이 너무 들어서 뭔가했는데 백준 15654번 N과M(8)에 조건하나 추가한 문제다.
기본적인 풀이 방법은 저 포스트에 담겨있다.

바뀐점은 (8)번은 중복을 허용하는 비내림차순이였는데 이번엔 중복은 허용하지 않는 비 내림차순이다.

입력
4 2
9 7 9 1

이런식으로 값이 들어오고

출력
1 7
1 9
7 9
9 9

이런식으로 출력된다.
보면 9가 중복으로 들어왔기 때문에 출력에서 [9, 9] 같은 수열은 나올 수 있다.
반복문에서 중복을 체크하는 boolean 배열과 현재 수열의 앞의 값을 체크하는 prev 변수조건만 붙여주면 된다.

정답 코드

const readline = require('readline');
const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout
});
let input = [];
let list = [];
let result = [];
let numbers = [];
let check = new Array(9).fill(false);
rl.on('line', function (line) {
  input.push(line);
})
  .on('close', async function () {
  // 답안 작성
  let answer =0;  
  let input_num = input[0].split(' ');
  numbers = input[1].split(' ').sort((a,b) => a-b);
  let n = input_num[0];
  let m = input_num[1];
  re(0,n,m,0);  
  console.log(result.join('\n'));  
  process.exit();
});
let re = function(cnt,n,m,start){
  if(cnt==m){    
    result.push(list.join(' '))
    return ;
  }
  let prev = -1;
  for(let i=start;i<n;i++){    
    if(!check[i]&&prev!=numbers[i]){      
      check[i] = true;
      prev = numbers[i];
      list[cnt] = numbers[i];
      re(cnt+1,n,m,i);      
      check[i] = false;
    }
  }
}