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

문제 : https://www.acmicpc.net/problem/2529

2529번: 부등호

나의 영원한 친구 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]]))