[Python/파이썬] 백준 17265번 나의 인생에는 수학과 함께

17265번: 나의 인생에는 수학과 함께

 

문제

세현이의 인생의 목표는 1분 1초 모든 순간 수학과 함께 살아가는 것이다. 그렇기 때문에 매일 수학을 생각하면서 살아가고 있다. 세현이는 밥을 먹을 때도 쌀알의 수를 계산하여 칼로리를 바로 계산하고 한걸음 한걸음 보폭을 계산하여 자신의 활동량을 확인하며 인생의 목표를 실행하며 살아가고 있다.  그런 세현이는 매일 학교를 가면서 지나가는 길에도 수학을 적용시키고 싶었다.

세현이네 집에서 학교까지 가는 길은 N x N 크기의 바둑판과 같다. 그리고 각 블록은 1x1 정사각형으로 구분 지을 수 있다. 세현이는 그 블록마다 숫자와 연산자가 존재한다고 생각해서 임의의 숫자와 연산자를 각 블록에 넣어 바둑판을 만들었다.

세현이는 학교에서 집으로 가는 경로에서 만나는 숫자와 연산자의 연산 결과의 최댓값과 최솟값을 구하려고 한다. 세현이는 항상 자신의 집 (1, 1)에서 학교 (N, N)까지 최단거리로 이동한다. 최단거리로 이동하기 위해서는 오른쪽과 아래쪽으로만 이동해야 한다.

위와 같이 N = 5 인 5 x 5 바둑판을 만들었다고 해보자.

그림 1의 경로를 통해 집(1, 1)에서 학교(N, N)까지 어떻게 연산이 되는지 확인해보자. 경로에서 만나는 연산자들의 우선순위는 고려하지 않는다.

1.  5 → × → 4 = 20
2.  20 → + → 5 = 25
3.  25 → ×→ 5 = 125
4.  125 → + → 2 = 127

그림 1은 최댓값을 가지는 경로이며 127이라는 최댓값을 얻을 수 있다.

그리고 위와 같이 연산하여 그림 2의 경로를 통해서 최솟값으로 3을 얻을 수 있다.

세현이는 이 길을 걸으면서 최댓값과 최솟값을 암산하다가 교통사고를 당해 현재 인하대학교 병원에 입원했다. 아픈 세현이를 위해 최댓값과 최솟값을 구해주자.

입력

첫 번째 줄에는 지도의 크기 N이 주어진다. (3 ≤ N ≤ 5, N은 홀수)

그 다음 N 줄에는 N개의 숫자 또는 연산자가 빈칸을 사이에 두고 주어지며, 세현이네 집 (1, 1)과 학교 (N, N)는 반드시 숫자로 주어진다.

그리고 숫자 다음에는 연산자, 연산자 다음에는 숫자가 나오도록 주어진다. 주어지는 숫자는 0이상 5이하의 정수, 연산자는 (‘+’, ‘-’, ‘*’) 만 주어진다.

 

출력

연산 결과의 최댓값과 최솟값을 하나의 공백을 두고 출력한다. 연산자 우선순위는 고려하지 않는다.

코드

n = int(input())
map_data = []
for _ in range(n):
    map_data.append(list(input().split()))

v = [[[int(1e9), -int(1e9)] for _ in range(n)] for _ in range(n)]

for i in range(n):
    for j in range(n):
        if i == 0 and j == 0:
            v[0][0][0] = int(map_data[0][0])
            v[0][0][1] = int(map_data[0][0])
            continue
        # 연산자
        if (i+j) % 2 == 1:
            if i == 0:
                v[i][j][0] = v[i][j-1][0]
                v[i][j][1] = v[i][j-1][1]
                continue
            if j == 0:
                v[i][j][0] = v[i-1][j][0]
                v[i][j][1] = v[i-1][j][1]
                continue
            v[i][j][0] = min(v[i-1][j][0], v[i][j-1][0])
            v[i][j][1] = max(v[i-1][j][1], v[i][j-1][1])
        # 숫자
        else:
            if i == 0:
                v[i][j][0] = eval(
                    str(v[i][j-1][0])+map_data[i][j-1]+map_data[i][j])
                v[i][j][1] = eval(
                    str(v[i][j-1][1])+map_data[i][j-1]+map_data[i][j])
                continue
            if j == 0:
                v[i][j][0] = eval(
                    str(v[i-1][j][0])+map_data[i-1][j]+map_data[i][j])
                v[i][j][1] = eval(
                    str(v[i-1][j][1])+map_data[i-1][j]+map_data[i][j])
                continue
            v[i][j][0] = min(eval(str(v[i-1][j][0])+map_data[i-1][j]+map_data[i][j]),
                             eval(str(v[i][j-1][0])+map_data[i][j-1]+map_data[i][j]))
            v[i][j][1] = max(eval(str(v[i-1][j][1])+map_data[i-1][j]+map_data[i][j]),
                             eval(str(v[i][j-1][1])+map_data[i][j-1]+map_data[i][j]))
print(v[n-1][n-1][1], v[n-1][n-1][0])

 

연산자와 숫자가 규칙적으로 나오기 때문에 규칙만 찾는다면 행과 열로 그 칸이 연산자인지 숫자인지 알 수 있다. 여기서는 행과 열 둘 중 하나만 홀수이면 그 칸은 연산자이다. 그러므로 행과 열의 합이 홀수라면 그 칸은 연산자임을 알 수 있다.

연산자 칸이라면 위쪽과 왼쪽을 비교하여 최소값과 최대값을 각각 구하여 그대로 갖고 온다.

숫자 칸이라면 문자열로 된 수식의 값을 계산해주는 eval 함수를 이용하여 최소값과 최대값을 구하여 저장해준다.