백준 10972번: 다음 순열 (C++)

 

 

1.문제

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

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

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

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

2.입력

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

3.출력

 

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

 

소스코드

#include <iostream>
#include<algorithm>
using namespace std;

int main()
{
	int arr[100001];
	int N,num =0;
	int index = 0;
	cin >> N;

	for (int i = 0; i < N; i++)
	{
		cin >> num;
		arr[i] = num;
	}

	for (int i = N-1; i >0; i--)
	{
		if (arr[i - 1] < arr[i])
		{
			index = i;
			break;
		}
	}

	if (index == 0)
	{
		cout << -1;
		return 0;
	}
	

	for (int i = N - 1; i >= index; i--)
	{
		if (arr[index - 1] < arr[i])
		{
			swap(arr[index-1], arr[i]);
			break;
		}
	}

	sort(arr+index, arr+N);

	for (int i = 0; i < N; i++)
	{
		cout << arr[i]<<" ";
	}
	return 0;
}

 

실행 결과

 

출처-    https://www.acmicpc.net/problem/10972