[완전탐색] 백준 #2529 부등호 - Python

🗂️ 문제

링크: https://www.acmicpc.net/problem/2529

2529번: 부등호

 

💡 접근법

1. 부등호 기호 앞뒤에 넣을 수 있는 숫자 0~9를 갖는 nums 리스트로 생성한다.
2. permutations를 활용하여 K+1자리수에 해당하는 순열들을 생성한다.
3. 부등호에 따라 값을 비교하고 비교 결과가 옳지 않을 경우에는 break문을 통해 for 문을 빠져나온다.
4. is_valid 변수를 활용하여 true일 경우, 즉 부등호에 따라 비교 결과가 옳은 순열에 대해서만 answers에 저장한다.
5. answers는 오름차순으로 정렬되어 있으므로 -1번째 인덱스와 0번째 인덱스에 해당하는 값을 출력한다.

 

😎 내 코드

import sys
from itertools import permutations

K = int(sys.stdin.readline())
C = sys.stdin.readline().split()

nums = [n for n in range(0, 10)]

answers = []
for p in permutations(nums, K+1):
    is_valid = True
    for i in range(len(p)-1): 
        if C[i] == '<':
            if p[i] > p[i+1]:
                is_valid = False
                break
        else:
            if p[i] < p[i+1]:
                is_valid = False
                break

    if is_valid:
        answers.append(p)

print(''.join(map(str, answers[-1])))
print(''.join(map(str, answers[0])))

 

✅ 정답 확인