문제 : https://www.acmicpc.net/problem/2529
2529번: 부등호
여러분은 제시된 부등호 관계를 만족하는 k+1 자리의 최대, 최소 정수를 첫째 줄과 둘째 줄에 각각 출력해야 한다. 단 아래 예(1)과 같이 첫 자리가 0인 경우도 정수에 포함되어야 한다. 모든 입력에 답은 항상 존재하며 출력 정수는 하나의 문자열이 되도록 해야 한다.
www.acmicpc.net
나의 영원한 친구 itertools
k의 범위가 2 ≤ k ≤ 9 이기 때문에 가능한 일이다.
코드 구현을 편하게 하기 위해 operator.lt, operator.gt 를 사용하여 <, > 를 연산자로 바꾼다.
그 후, 숫자 리스트를 만들어 permutation을 통해 전부 돌려 확인한다.
import sys
import operator
from itertools import combinations,permutations,combinations_with_replacement
n = int(sys.stdin.readline())
tmp = [c for c in sys.stdin.readline().split()]
ops = []
ret=[]
for s in tmp:
if s=="<":
ops.append(operator.lt)
else:
ops.append(operator.gt)
num = [i for i in range(10)]
for c in permutations(num, n+1):
yesman = True
for i in range(n):
if not ops[i](c[i],c[i+1]):
yesman = False
if yesman:
ret.append(c)
print("".join([str(i) for i in ret[-1]]))
print("".join([str(i) for i in ret[0]]))
'취준 > 백준' 카테고리의 다른 글
[백준][Python][15686][브루트포스] 치킨 배달 (0) | 2020.03.03 |
---|---|
[백준][Python][1697][BFS] 숨바꼭질 (0) | 2020.03.02 |
[백준][Python][2294][DP] 동전 2 (0) | 2020.02.29 |
[백준][Python][11054][DP]가장 긴 바이토닉 부분 수열 (0) | 2020.02.28 |
[백준][Python][2293][DP] 동전 1 (0) | 2020.02.28 |