문제
독일 로또는 {1, 2, ..., 49}에서 수 6개를 고른다.
로또 번호를 선택하는데 사용되는 가장 유명한 전략은 49가지 수 중 k(k>6)개의 수를 골라 집합 S를 만든 다음 그 수만 가지고 번호를 선택하는 것이다.
예를 들어, k=8, S={1,2,3,5,8,13,21,34}인 경우 이 집합 S에서 수를 고를 수 있는 경우의 수는 총 28가지이다. ([1,2,3,5,8,13], [1,2,3,5,8,21], [1,2,3,5,8,34], [1,2,3,5,13,21], ..., [3,5,8,13,21,34])
집합 S와 k가 주어졌을 때, 수를 고르는 모든 방법을 구하는 프로그램을 작성하시오.
입력
입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있다. 첫 번째 수는 k (6 < k < 13)이고, 다음 k개 수는 집합 S에 포함되는 수이다. S의 원소는 오름차순으로 주어진다.
입력의 마지막 줄에는 0이 하나 주어진다.
출력
각 테스트 케이스마다 수를 고르는 모든 방법을 출력한다. 이때, 사전 순으로 출력한다.
각 테스트 케이스 사이에는 빈 줄을 하나 출력한다.
예제 입력 1
7 1 2 3 4 5 6 7
8 1 2 3 5 8 13 21 34
0
Plain Text
복사
예제 출력 1
1 2 3 4 5 6
1 2 3 4 5 7
1 2 3 4 6 7
1 2 3 5 6 7
1 2 4 5 6 7
1 3 4 5 6 7
2 3 4 5 6 7
1 2 3 5 8 13
1 2 3 5 8 21
1 2 3 5 8 34
1 2 3 5 13 21
1 2 3 5 13 34
1 2 3 5 21 34
1 2 3 8 13 21
1 2 3 8 13 34
1 2 3 8 21 34
1 2 3 13 21 34
1 2 5 8 13 21
1 2 5 8 13 34
1 2 5 8 21 34
1 2 5 13 21 34
1 2 8 13 21 34
1 3 5 8 13 21
1 3 5 8 13 34
1 3 5 8 21 34
1 3 5 13 21 34
1 3 8 13 21 34
1 5 8 13 21 34
2 3 5 8 13 21
2 3 5 8 13 34
2 3 5 8 21 34
2 3 5 13 21 34
2 3 8 13 21 34
2 5 8 13 21 34
3 5 8 13 21 34
Plain Text
복사
My solution
k=8, S={1,2,3,5,8,13,21,34}인 경우 이 집합 S에서 수를 고를 수 있는 경우의 수는 총 28가지이다. ([1,2,3,5,8,13], [1,2,3,5,8,21], [1,2,3,5,8,34], [1,2,3,5,13,21], ..., [3,5,8,13,21,34])
TC별로 한줄의 공백이 필요하므로 빈문자열을 하나 삽입해서 split(’\n’)으로 넣어줬다.
마지막 TC의 경우에는 pop()으로 빈문자열을 없애서 공백을 제거했다.
...
DFS([], 0, 0);
result.push('');
}
result.pop(); // 맨 뒤에 붙은 빈 문자열은 제거
console.log(result.join('\n'));
return result.join('\n');
JavaScript
복사
DFS로 조합을 구하면 된다.
초기 배열 []를 전달해서 함수 외부에 배열에 대한 변수를 남기지 않도록 했다.
depth가 6에 도착하면 6개를 뽑았다는 것으로 이때마다 result에 추가한다.]
조합은 이전 인덱스를 참조할 수 없도록(중복된 숫자로 구성되지 않도록) idx + 1로 재귀호출을 수행하면 된다.
const visited = new Array(K).fill(false);
const DFS = (comb, idx, depth) => {
if (depth === 6) {
result.push(comb.join(' '));
return;
}
for (let i = idx; i < K; i++) {
if (!visited[i]) {
visited[i] = true;
comb[depth] = S[i];
DFS(comb, i + 1, depth + 1);
visited[i] = false;
comb.pop();
}
}
};
DFS([], 0, 0);
JavaScript
복사
스터디원 풀이
조합을 풀이할 때 i = idx;로 주기 때문에 visited 배열은 필요하지 않을 것 같다. 배열에 담아서 includes로 처리할 수도 있을듯
const DFS = (comb, idx, depth) => {
if (depth === 6) {
result.push(comb.join(' '));
return;
}
for (let i = idx; i < K; i++) {
if (!visited[i]) {
visited[i] = true;
comb[depth] = S[i];
DFS(comb, i + 1, depth + 1);
visited[i] = false;
}
}
};
JavaScript
복사
Code
const solution = function (i) {
const TC = i.toString().trim().split('\n');
TC.pop(); // 마지막에 들어온 0 제거
const result = [];
for (const tc of TC) {
const [K, ...S] = tc.split(' ').map(Number);
const visited = new Array(K).fill(false);
const DFS = (comb, idx, depth) => {
if (depth === 6) {
result.push(comb.join(' '));
return;
}
for (let i = idx; i < K; i++) {
if (!visited[i]) {
visited[i] = true;
comb[depth] = S[i];
DFS(comb, i + 1, depth + 1);
visited[i] = false;
comb.pop();
}
}
};
DFS([], 0, 0);
result.push('');
}
result.pop(); // 맨 뒤에 붙은 빈 문자열은 제거
console.log(result.join('\n'));
return result.join('\n');
};
describe('로또(조합)', () => {
it('TC1', () => {
expect(
solution(`7 1 2 3 4 5 6 7
8 1 2 3 5 8 13 21 34
0`)
).toStrictEqual(`1 2 3 4 5 6
1 2 3 4 5 7
1 2 3 4 6 7
1 2 3 5 6 7
1 2 4 5 6 7
1 3 4 5 6 7
2 3 4 5 6 7
1 2 3 5 8 13
1 2 3 5 8 21
1 2 3 5 8 34
1 2 3 5 13 21
1 2 3 5 13 34
1 2 3 5 21 34
1 2 3 8 13 21
1 2 3 8 13 34
1 2 3 8 21 34
1 2 3 13 21 34
1 2 5 8 13 21
1 2 5 8 13 34
1 2 5 8 21 34
1 2 5 13 21 34
1 2 8 13 21 34
1 3 5 8 13 21
1 3 5 8 13 34
1 3 5 8 21 34
1 3 5 13 21 34
1 3 8 13 21 34
1 5 8 13 21 34
2 3 5 8 13 21
2 3 5 8 13 34
2 3 5 8 21 34
2 3 5 13 21 34
2 3 8 13 21 34
2 5 8 13 21 34
3 5 8 13 21 34`);
});
});
JavaScript
복사
