[ 백준 15650 / Node.js ] N과 M (2)

https://www.acmicpc.net/problem/15650

15650번: N과 M (2)

 

문제

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

• 1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
• 고른 수열은 오름차순이어야 한다.

입력

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

출력

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

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

예제 입력 1 복사

3 1

예제 출력 1 복사

1
2
3

예제 입력 2 복사

4 2

예제 출력 2 복사

1 2
1 3
1 4
2 3
2 4
3 4

예제 입력 3 복사

4 4

예제 출력 3 복사

1 2 3 4

풀이

고른 수열은 오름차순이어야 한다

이 문제 역시 N과 M (1)과 마찬가지로 DFS로 해결이 가능하다.

N과 M (1)은 모든 경우의 수를 위해, 재귀를 위한 for loop에서 index를 항상 맨 처음부터 다시 돌았다면,
N과 M (2)에서는 중복없이 M개의 고른 수열을 오름차순으로 구하기 위해, 지난 숫자는 제외하고 for loop을 진행한다.

그래서 for loop에서 1이 아닌 start부터 시작한다.

 

let fs = require('fs');
let [N, M] = fs.readFileSync('/dev/stdin').toString().trim().split(' ').map(Number);

const arr = Array(M).fill('');
const visited = Array(N + 1).fill(false);
dfs(0, 1);

function dfs(lev, start) {
  if (lev === M) {
    console.log(arr.join(' '));
    return;
  }
  for (let i = start; i <= N; i++) {
    if (visited[i]) {
      continue;
    }
    arr[lev] = i;
    visited[i] = true;
    dfs(lev + 1, i);
    visited[i] = false;
  }
}