취준/백준
[백준][Python][2529][Greedy...?] 부등호
puff
2020. 3. 1. 10:28
문제 : 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]]))