백준 15663 N과 M(9)

문제

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

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 = new Set();
  let visited = new Array(arr.length).fill(0);
  let tmp = [];
  const dfs = (L, s) => {
    if (L === m) {
      answer.add(tmp.join(" "));
    } else {
      for (let i = 0; i < arr.length; i++) {
        if (!visited[i]) {
          visited[i] = 1;
          tmp.push(arr[i]);
          dfs(L + 1, s + 1);
          tmp.pop();
          visited[i] = 0;
        }
      }
    }
  };
  dfs(0, 0);
  return [...answer].join("\n");
};

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

• set을 이용하고 같은 수 사용이 안되는 순열
• set을 이용해 배열에 겹치는 요소들이 있어 중복되는 수열을 제거해준다.