[백준 15656] N과 M(7) (C++/Python)

- N과 M (1) : 중복 수열 X / 숫자 중복 X
- N과 M (2) : 중복 수열 X / 숫자 중복 X / 오름차순
- N과 M (3) : 중복 수열 X / 숫자 중복 O
- N과 M (4) : 중복 수열 X / 숫자 중복 O / 비내림차순
~~아래부터는 사용자로부터 직접 입력받음~~
- N과 M (5) : 중복 수열 X / 사전 증가순 
- N과 M (6) : 중복 수열 X / 사전 증가순 / 오름차순
- N과 M (7) : 중복 수열 X / 사전 증가순 / 같은 수 여러번 O

 

---

 

C++)

#include <iostream>
#include <algorithm>
#include <vector>
#define MAX_NUM 10
using namespace std;

int n, m;
vector<int> v;
int ans[MAX_NUM];		// 정답 가진 배열

// cnt == 현재 수열 길이
// cur == 탐색 시작 위치
void suyeol(int cnt) {
	if (cnt == m) {
		for (int i = 0; i < m; i++)
			cout << ans[i] << " ";
		cout << "\n";
		return;
	}

	for (int i = 0; i < n; i++) {
		ans[cnt] = v.at(i);
		suyeol(cnt + 1);
	}
}

int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);

	cin >> n >> m;

	int temp;
	for (int i = 0; i < n; i++) {
		cin >> temp;
		v.push_back(temp);
	}

	// 사전순이니까 오름차순 정렬
	sort(v.begin(), v.end());

	suyeol(0);

	return 0;
}

 

 

 

Python)

import sys

read = sys.stdin.readline

n, m = map(int, read().rstrip().split())
ans = [0] * m
visited = [False] * n

arr = list(map(int, read().rstrip().split()))

def suyeol(cnt):
    if cnt == m:
        print(' ' .join(map(str, ans)))
        return
    
    for i in range(n):
        ans[cnt] = arr[i]
        suyeol(cnt+1)
    
arr.sort()
suyeol(0)