[백준] 10972 다음순열

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

10972번: 다음 순열

 

from itertools  import permutations

n = int(input())
array = list(map(int, input().split()))
perm = list(permutations(range(1, n+1), n))

for i in range(1, len(perm)):
    if list(perm[i-1]) == array:
        print(' '.join(map(str, perm[i])))
        break
    if i == len(perm)-1:
        print('-1')

메모리 초과 오류가 발생 한다 permutaion을 이용하면 메모리를 많이 차지 하기 때문에 다른 방법을 구상해야 한다. 

 

n = int(input())
array = list(map(int, input().split()))

for i in range(n - 1, 0, -1):  
    if array[i - 1] < array[i]:  
        for j in range(n - 1, 0, -1):  
            if array[i - 1] < array[j]: 
                array[i - 1], array[j] = array[j], array[i - 1]  
                array = array[:i] + sorted(array[i:])  
                print(*array)  
                exit()  

print(-1)