백준 15654번: N과 M (5) - javascript

백준 15654번: N과 M (5)

문제 설명

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

• N개의 자연수 중에서 M개를 고른 수열

입력

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

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

출력

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

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

접근

이미 포스팅했던 백준 15649번 N과M의 응용 버전이다.
기본적인 풀이 방법은 저 포스팅을 보면된다.

그런데 이번에는 1부터 N까지의 자연수가 아닌 N개의 자연수이다.
이러면 기본 풀이에서 둘째 줄에 입력받는 N개의 자연수들을 보관하는 전역변수가 하나 더 필요하다.
그 다음에는 기존에는 list[cnt] 에 i를 추가하던 부분을 전역변수 배열의 i번재 항을 추가하는 방식으로 바꾸면 된다.
ps 당연히 수열은 사전 순으로 증가한다고 했으니 맨처음 입력받을때 정렬해서 넣어줘야 한다.

정답 코드

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,k){
  if(cnt==m){
    result.push(list.join(' '))
    return 1;
  }
  for(let i=0;i<n;i++){
    if(!check[i] ){
      check[i] = true;
      list[cnt] = numbers[i];
      re(cnt+1,n,m,k);
      check[i] = false;
    }
  }
  return 1;
}