10972 다음 순열

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

10972번: 다음 순열

다음 순열 성공

시간 제한메모리 제한제출정답맞은 사람정답 비율

1 초

256 MB

10799

4691

3393

45.042%

문제

1부터 N까지의 수로 이루어진 순열이 있다. 이때, 사전순으로 다음에 오는 순열을 구하는 프로그램을 작성하시오.

사전 순으로 가장 앞서는 순열은 오름차순으로 이루어진 순열이고, 가장 마지막에 오는 순열은 내림차순으로 이루어진 순열이다.

N = 3인 경우에 사전순으로 순열을 나열하면 다음과 같다.

• 1, 2, 3
• 1, 3, 2
• 2, 1, 3
• 2, 3, 1
• 3, 1, 2
• 3, 2, 1

입력

첫째 줄에 N(1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄에 순열이 주어진다.

출력

첫째 줄에 입력으로 주어진 순열의 다음에 오는 순열을 출력한다. 만약, 사전순으로 마지막에 오는 순열인 경우에는 -1을 출력한다.

 

 

<풀이>

1. 문제의 조건의 뜻과, 예시 출력을 보고 의미를 파악한다.

 [사전 순으로 가장 앞서는 순열은 오름차순으로 이루어진 순열이고,

                  가장 마지막에 오는 순열은 내림차순]

 

1,2,3,4

1,2,4,3

1,3,2,4

....(생략)

 

각 순열의 항을 a(1), a(2), a(3) 이라고하면

조건과 예시의 뜻은 a(1) <a(2) < a(3) 순서로 순열을 이룬다.

 

가장 마지막 숫자는 입력한 수의 최대 값이 될 것이다.

ex ) 1,2,3,4   -> a(2) -> a(3) -> ......... a(x) = 4,3,2,1

 

 

2. 예제 다음의 숫자에서 조건의 사전순에 맞는 다음 순열을 찾기.

 

i - 1과 k의 수를 바꿔줍니다.

 

그럼 4, 1, 5, 3, 2 가 됩니다.

 

 

i 번째 부터 마지막 인덱스 까지 모든 수를 바꿔줍니다.

 

ex) 4 1 5 3 2 -----> 4 1 5 2 3

 

정답 = 4 1 5 2 3

#include<stdio.h>
#include<algorithm>
#include<stack>
bool next_permutation(int* a,int *index);
int n;

using namespace std;
int main(void)
{
	
	scanf("%d", &n);
	int index = 0;
	int arr[10001] = { 0, };
	
	for (int i = 0; i < n; ++i)
	{
		scanf("%d", &arr[i]);
	}

	if (!next_permutation(arr, &index))
	{
		printf("-1\n");
		return 0;
	}
	stack<int> s;

	for (int i = index; i < n; ++i)
	{
		s.push(arr[i]);
	}
	for (int i = index; i < n; ++i)
	{
		arr[i] = s.top();
		s.pop();
	}
	for (int i = 0; i < n; ++i)
	{
		printf("%d ", arr[i]);
	}
	printf("\n");
	return 0;
}
bool next_permutation(int* a, int* index)
{
	int next = 0;
	int i_num = 0;
	int k_num = 0;

	for (int i = 0; i < n-1; ++i)
	{
		next = i + 1;
		if (a[i] < a[next])
		{
			i_num = next;
		}
	}
	for (int k = i_num - 1; k < n - 1; ++k)
	{
		next = k + 1;
		if (a[i_num-1] < a[next])
		{
			k_num = next;
		}
		if (k_num == 0) 
			return false;
	}
	
	int temp = a[k_num];
	a[k_num] = a[i_num - 1];
	a[i_num - 1] = temp;

	*index = i_num;
	return true;
}