백준: 2529 부등호

문제

2529번: 부등호

 

문제 풀이

 부등호가 2개가 주어진다면 부등호 사이에 들어갈 수 있는 숫자는 3개이다. 따라서 수가 중복되지 않고 수의 순서에 따라 다른 경우의 수가 되므로 이는 순열을 구하는 문제이다. 이전의 N과 M (1)과 다른점이 있다면, 입력된 부등호의 조건을 만족하는지에 대해 확인 할 필요가 있다.

 

 

코드

def available(i, j, op):
    if op == '<':
        return i < j
    return i > j


def dfs(depth):
    global answer

    if depth == k + 1:
        '''
        가장 처음에 찾는 값이 MIN
        가장 마지막에 찾는 값이 MAX
        '''
        if not answer[MIN]:
            answer[MIN] = [*check]
        else:
            answer[MAX] = [*check]
        return

    for i in range(10):
        if i not in check:
            if not depth or available(check[depth - 1], i, op[depth - 1]):
                check[depth] = i
                dfs(depth + 1)
                check[depth] = -1


if __name__ == "__main__":
    MIN, MAX = 0, 1
    answer = [[] for _ in range(2)]
    k = int(input())
    op = input().split()
    nums = [num for num in range(10)]
    check = [-1] * (k + 1)
    dfs(0)

    for num in answer[MAX]:
        print(num, end='')
    print()
    for num in answer[MIN]:
        print(num, end='')

 로직을 정확히 작성하였음에도 불구하고, 문제에서 채점 실패를 하는 경우가 발생하였다. 30분 동안 코드도 보고 반례들도 입력해보았는데, `check`를 해제할 때 0인덱스와 안겹치게 -1로 초기화해주어야 하는데 0으로 초기화하다보니 실패가 발생하였다.