[python/파이썬] 백준 1890 점프

문제

N×N 게임판에 수가 적혀져 있다. 이 게임의 목표는 가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 규칙에 맞게 점프를 해서 가는 것이다.

각 칸에 적혀있는 수는 현재 칸에서 갈 수 있는 거리를 의미한다. 반드시 오른쪽이나 아래쪽으로만 이동해야 한다. 0은 더 이상 진행을 막는 종착점이며, 항상 현재 칸에 적혀있는 수만큼 오른쪽이나 아래로 가야 한다. 한 번 점프를 할 때, 방향을 바꾸면 안 된다. 즉, 한 칸에서 오른쪽으로 점프를 하거나, 아래로 점프를 하는 두 경우만 존재한다.

가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 규칙에 맞게 이동할 수 있는 경로의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 게임 판의 크기 N (4 ≤ N ≤ 100)이 주어진다. 그 다음 N개 줄에는 각 칸에 적혀져 있는 수가 N개씩 주어진다. 칸에 적혀있는 수는 0보다 크거나 같고, 9보다 작거나 같은 정수이며, 가장 오른쪽 아래 칸에는 항상 0이 주어진다.

출력

가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 문제의 규칙에 맞게 갈 수 있는 경로의 개수를 출력한다. 경로의 개수는 263-1보다 작거나 같다.

예제 입력 1 복사

4
2 3 3 1
1 2 1 3
1 2 3 1
3 1 1 0

예제 출력 1 복사

3

 

 

 

[소스 코드1] - 맞았습니다!

 

n = int(input())

arr = []
t_stack = []
count = 0

for i in range(n):
    tmp = list(map(int, input().split()))
    arr.append(tmp)
    
dp = [[0 for j in range(n)] for i in range(n)]

dp[0][0] = 1

for i in range(n):
    for j in range(n):
        if i == n-1 and j == n-1:
            print(dp[i][j])
            break
        tmp = arr[i][j] #현재 칸에 써있는 숫자
    
        if tmp < n - i: #숫자만큼 아래로 이동할 수 있다면
            dp[i+tmp][j] += dp[i][j]
        if tmp < n- j: #숫자만큼 오른쪽으로 이동할 수 있다면
            dp[i][j+tmp] += dp[i][j]

 

[소스 코드2] - 틀렸습니다!

 

튜플과 스택(리스트)을 이용해서 처음에 풀어봤던 코드이다.

예시에 있는 데이터를 넣었을 때 답이 제대로 나오고 동작도 잘하는 것 같은데

틀린 문제이다. 어느 부분이 문제인지 잘 모르겠어서 첨부해본다.

n = int(input())
arr = []
t_stack = []
count = 0

for i in range(n):
    tmp = list(map(int, input().split()))
    arr.append(tmp)
    
t_stack.append((arr[0][0],0))
t_stack.append((0,arr[0][0]))

for i in range(10000):
    if len(t_stack)==0:
        break
    el = t_stack.pop()
    
    tmp = arr[el[0]][el[1]]
    if tmp == 0:
        count += 1
        continue
    if tmp < n - el[0]:
        t_stack.append((el[0]+tmp, el[1]))
    if tmp < n-el[1]: 
        t_stack.append((el[0], el[1]+tmp))
    

print(count)