백준 15655 N과 M(6)

문제

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

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

제한 사항

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

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

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

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

입출력 예

풀이

let input = require('fs').readFileSync('/dev/stdin').toString().trim().split('\n');
let [n,m] = [...input.shift().split(" ").map(e => +e)]
let arr = input.shift().split(" ").map(e => +e);

const solution = (arr, m) => {
  arr.sort((a, b) => a - b);
  let answer = [];
  let visited = new Array(arr.length).fill(0);
  let tmp = [];
  const dfs = (index, cnt) => {
    if (cnt === m) {
      answer.push(tmp.join(" "));
    } else {
      for (let i = index; i < arr.length; i++) {
        if (!visited[i]) {
          visited[i] = 1;
          tmp.push(arr[i]);
          dfs(i, tmp.length);
          visited[i] = 0;
          tmp.pop();
        }
      }
    }
  };
  dfs(0, 0);
  return answer.join("\n");
};

console.log(solution(arr, m));

• 같은 수를 허용하지 않는 조합