백준 15656 N과 M(7)

문제

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

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


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

같은 수 사용 가능한 순열

const sol = (arr, m) => {
  const answer = [];
  //   const visited = new Array(arr.length).fill(0);
  let tmp = [];

  const dfs = (L) => {
    if (m === tmp.length) {
      answer.push(tmp.join(" "));
    } else {
      for (let i = 0; i < arr.length; i++) {
        tmp.push(arr[i]);
        dfs(L + 1);
        tmp.pop();
      }
    }
  };
  dfs(0);
  return answer;
};

console.log(sol([9, 8, 7, 1], 2).sort());