백준 15655번: N과 M (6) - javascript

백준 15655번: N과 M (6)

문제 설명

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

• N개의 자연수 중에서 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.

입력

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

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

출력

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

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

접근

이번 문제는 백준 15654번 N과M(5)의 응용 버전이다.
기본적인 풀이 방법은 저 포스팅에서 시작된다.

거기서 약간 다른 점이라면 각 수열은 내부적으로 오름차순이어야 한다는 점이다.
그래서 현재 값과 비교할수 있는 비교값을 유지시켜서 오름차순을 유지하려고 했는데 틀렸다고 나왔다.

let re = function(cnt,n,m,k){
  if(cnt==m){
    result.push(list.join(' '))
    return ;
  }
  for(let i=0;i<n;i++){
    if(!check[i] && k<numbers[i] ){
      let tmp = k;
      k = numbers[i];
      check[i] = true;
      list[cnt] = numbers[i];
      re(cnt+1,n,m,k);
      k = tmp;
      check[i] = false;
    }
  }  
}

보면 초기값 0을 가지는 k라는 변수를 추가적으로 두어서 오름차순을 유도하고자 했는데, 채점에서는 매번 7%에서 틀렸다.
그래서 어차피 입력받은 자연수들을 오름차순으로 정렬했으니 반복문의 시작값을 한 번 재귀할때마다 1씩 올려주는 방식으로 바꿨더니 코드가 제대로 동작했다…

반례를 알고 싶었는데 예제는 두 코드다 똑같이 돌아가서 찾지는 못했다. 아마 주소에 의한 객체 참조 문제 아니였을까..

정답 코드

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 ;
  }
  for(let i=start;i<n;i++){
    if(!check[i]){      
      check[i] = true;
      list[cnt] = numbers[i];
      re(cnt+1,n,m,i+1);      
      check[i] = false;
    }
  }
}