백준 15665 N과 M (11)

문제

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

• N개의 자연수 중에서 M개를 고른 수열
• 같은 수를 여러 번 골라도 된다.

제한 사항

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

• 둘째 줄에 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 = new Set();
  let tmp = [];
  const dfs = (cnt) => {
    if (cnt === m) {
      answer.add(tmp.join(" "));
    } else {
      for (let i = 0; i < arr.length; i++) {
        tmp.push(arr[i]);
        dfs(cnt + 1);
        tmp.pop();
      }
    }
  };
  dfs(0);
  return [...answer].join("\n");
};
console.log(solution(arr, m));

• 배열의 같은 요소가 있어 중복되는 요소를 제거한 순열