[BOJ/백준] 2992번 PYTHON

유승한·2024년 1월 18일
0

algorithm

목록 보기
4/6

문제

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

유형

브루트포스

배운 점

itertools 모듈의 permutations를 이용해 해결했다.
이때 combinations, permutations 함수는 tuple 형태를 반환하는 것이 아니며 자료형의 객체를 반환한다.
또한 set함수를 이용하여 중복을 제거해줬다.

코드

from itertools import permutations

def combinations(string):
    number_str = string
    perm = permutations(number_str)
    result = list(set([(''.join(p))for p in perm]))

    return sorted(result)

string = input()
comb = combinations(string)
index = comb.index(string)+1
if index < len(comb):
    print(comb[index])
else:
    print(0)

0개의 댓글

관련 채용 정보