[BOJ][⚪2][백준#02529] 부등호

문제 출처

BAEKJOON Online Judge #2529

문제

두 종류의 부등호 기호 ‘<’와 ‘>’가 k개 나열된 순서열 A가 있다. 우리는 이 부등호 기호 앞뒤에 서로 다른 한 자릿수 숫자를 넣어서 모든 부등호 관계를 만족시키려고 한다. 예를 들어, 제시된 부등호 순서열 A가 다음과 같다고 하자.  A ⇒ < < < > < < > < > 부등호 기호 앞뒤에 넣을 수 있는 숫자는 0부터 9까지의 정수이며 선택된 숫자는 모두 달라야 한다. 아래는 부등호 순서열 A를 만족시키는 한 예이다.  3 < 4 < 5 < 6 > 1 < 2 < 8 > 7 < 9 > 0 이 상황에서 부등호 기호를 제거한 뒤, 숫자를 모두 붙이면 하나의 수를 만들 수 있는데 이 수를 주어진 부등호 관계를 만족시키는 정수라고 한다. 그런데 주어진 부등호 관계를 만족하는 정수는 하나 이상 존재한다. 예를 들어 3456128790 뿐만 아니라 5689023174도 아래와 같이 부등호 관계 A를 만족시킨다.  5 < 6 < 8 < 9 > 0 < 2 < 3 > 1 < 7 > 4 여러분은 제시된 k개의 부등호 순서를 만족하는 (k+1)자리의 정수 중에서 최댓값과 최솟값을 찾아야 한다. 앞서 설명한 대로 각 부등호의 앞뒤에 들어가는 숫자는 { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }중에서 선택해야 하며 선택된 숫자는 모두 달라야 한다. 

입력

첫 줄에 부등호 문자의 개수를 나타내는 정수 k가 주어진다. 그 다음 줄에는 k개의 부등호 기호가 하나의 공백을 두고 한 줄에 모두 제시된다. k의 범위는 2 ≤ k ≤ 9 이다. 

출력

여러분은 제시된 부등호 관계를 만족하는 k+1 자리의 최대, 최소 정수를 첫째 줄과 둘째 줄에 각각 출력해야 한다. 단 아래 예(1)과 같이 첫 자리가 0인 경우도 정수에 포함되어야 한다. 모든 입력에 답은 항상 존재하며 출력 정수는 하나의 문자열이 되도록 해야 한다. 

예제

예제 1

입력

2
< >

출력

897
021

예제 2

입력

9
> < < < > > > < <

출력

9567843012
1023765489

My Sol

def dfs(i):
    global stack
    if i==N+1:
        if not check(stack[-2], ops[-1], stack[-1]): return
        global ansLst
        txt = ''.join(list(map(str, stack)))
        ansLst.append(txt)
        return

    if i > 1:
        if not check(stack[i-2], ops[i-2], stack[i-1]):
            return

    for k in range(10):
        if visited[k]: continue
        visited[k] = 1
        stack.append(k)
        dfs(i+1)
        visited[k] = 0
        stack.pop()


def check(n1, op, n2):
    if op=='<':
        return n1<n2
    return n1>n2


N = int(input())
ops = list(input().split())
arr = list(range(1, 10))
visited = [0]*10
stack = []
ansLst = []
dfs(0)

print(ansLst[-1])
print(ansLst[0])

백트래킹을 이용하는 방식이었다. DFS를 이용해 0부터 9까지의 모든 가능한 조합을 만들어가는데, 부등호를 매번 체크하여 만약 부등호의 관계가 만족되지 않는다면 이후의 트리는 더 진행하지 않고 가지치기하는 방식으로 시간을 줄였다.

i가 N+1까지 가면, 마지막 부등호만 체크해서 이마저도 만족한다면 전역 배열 변수인 ansLst에 넣어준다. for문을 이용해 순서대로 조합을 하기 때문에 가장 처음 들어간 것이 가장 작은 수가 되고, 마지막으로 들어간 것이 가장 큰 수가 되기 때문에 모든 DFS조회가 끝나면 끝값과 첫값을 출력해주면 되겠다.

결과

맞았습니다!!