본문 바로가기

취준/백준

[백준][Python][2529][Greedy...?] 부등호

문제 : 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]]))