백준 15657번: N과 M (9) - javascript

백준 15657번: N과 M (9)

문제 설명

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

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

입력

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

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

출력

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

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

접근

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

(6)번과 다르게, 아니 여태까지 나온 n과m 시리즈와 다르게 주어진 N개의 수에 중복된 수가 들어온다.

입력
4 2
9 7 9 1

이런식으로 값이 들어오고

출력
1 7
1 9
7 1
7 9
9 1
9 7
9 9

이런식으로 출력이 되야한다.
생각보다 간단하게 풀어봤는데 맞았다.
내 코드의 경우 맨처음 N개의 수를 입력받을 때 한번 순서대로 정렬해준다.
그러므로 중복된 수가 있으면 서로 붙어있을 것이다. 그 중복된수가 몇 개 일지라 해도 말이다!
그래서 check 해주는 boolean 배열체크 말고 추가로 현재 i행보다 -1만큼 앞의 값을 저장해두고 현재 i 행값과 다르면 list를 만들도록 조건을 추가해줬다.

정답 코드

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();
});
//N개의 자연수, 차순구별x 중복수열 x 중복 x
let re = function(cnt,n,m,start){
  if(cnt==m){    
    result.push(list.join(' '))
    return ;
  }
  //N개의 자연수가 들어오니깐 초기값은 -1이다.
  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,start);      
      check[i] = false;
    }
  }
}