백준 1039번 : 교환 (c++)

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

1039번: 교환

문제

0으로 시작하지 않는 정수 N이 주어진다. 이때, M을 정수 N의 자릿수라고 했을 때, 다음과 같은 연산을 K번 수행한다.
1 ≤ i < j ≤ M인 i와 j를 고른다. 그 다음, i번 위치의 숫자와 j번 위치의 숫자를 바꾼다. 이때, 바꾼 수가 0으로 시작하면 안 된다.
위의 연산을 K번 했을 때, 나올 수 있는 수의 최댓값을 구하는 프로그램을 작성하시오.

입력
첫째 줄에 정수 N과 K가 주어진다. N은 1,000,000보다 작거나 같은 자연수이고, K는 10보다 작거나 같은 자연수이다.

출력
첫째 줄에 문제에 주어진 연산을 K번 했을 때, 만들 수 있는 가장 큰 수를 출력한다. 만약 연산을 K번 할 수 없으면 -1을 출력한다.

예제 입력
16375 1
예제 출력 
76315

예제 입력
132 3
예제 출력
312

#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>

using namespace std;

string N;
int K;
int dp[1000001][11];

int dfs(string a, int depth) {
	if (depth == K) {
		return stoi(a);
	}
	int& ret = dp[stoi(a)][depth];
	if (ret != -1) return ret;

	for (int i = 0; i < a.size(); i++)
	{
		for (int j = i + 1; j < a.size(); j++)
		{
			if (a[j] == '0' && i == 0)    continue;
			swap(a[i], a[j]);
			ret = max(ret, dfs(a, depth + 1));
			swap(a[i], a[j]);
		}
	}
	return ret;
}

int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);

	cin >> N >> K;
	memset(dp, -1, sizeof(dp));
	cout << dfs(N, 0);

	return 0;
}