문제. 프로그래머스 등굣길
문제 설명
계속되는 폭우로 일부 지역이 물에 잠겼습니다. 물에 잠기지 않은 지역을 통해 학교를 가려고 합니다. 집에서 학교까지 가는 길은 m x n 크기의 격자모양으로 나타낼 수 있습니다.
아래 그림은 m = 4, n = 3 인 경우입니다.
가장 왼쪽 위, 즉 집이 있는 곳의 좌표는 (1, 1)로 나타내고 가장 오른쪽 아래, 즉 학교가 있는 곳의 좌표는 (m, n)으로 나타냅니다.
격자의 크기 m, n과 물이 잠긴 지역의 좌표를 담은 2차원 배열 puddles이 매개변수로 주어집니다. 오른쪽과 아래쪽으로만 움직여 집에서 학교까지 갈 수 있는 최단경로의 개수를 1,000,000,007로 나눈 나머지를 return 하도록 solution 함수를 작성해주세요.
제한사항
- 격자의 크기 m, n은 1 이상 100 이하인 자연수입니다.
- m과 n이 모두 1인 경우는 입력으로 주어지지 않습니다.
- 물에 잠긴 지역은 0개 이상 10개 이하입니다.
- 집과 학교가 물에 잠긴 경우는 입력으로 주어지지 않습니다.
입출력 예
m | n | puddles | return |
4 | 3 | [[2, 2]] | 4 |
입출력 예 설명
출처 : 프로그래머스 등굣길
나의 풀이 in Python
def solution(m, n, puddles):
location = [[0]*(m+1) for _ in range(n+1)]
location[1][1] = 1
for i in range(1,n+1):
for j in range(1,m+1):
if i ==1 and j ==1:
continue
if [j,i] in puddles:
location[i][j] = 0
else:
location[i][j] += location[i-1][j] +location[i][j-1]
return location[-1][-1] % 1000000007
문제의 핵심은 좌표당 경로의 개수를 구해가며 최종적으로 학교가 있는 곳의 좌표 (m,n)의 경로 개수를 구하는 것입니다.
문제 설명
1. 좌표당 경로의 개수를 기입하기 위해 location 변수를 생성
파이썬의 인덱싱은 0부터이므로 (m,n)이 아닌 (m+1,n+1)으로 설정함.
2. 시작 부분은 1의 값을 분배, puddles가 있을 경우, 경로의 개수를 0으로 설정함.
3. 가로와 세로로만 이동이 가능하기에 location[i][j]는 location[i-1][j] + location[i][j-1] 이라는 식을 따름.
해당 문제의 난이도는 어렵지 않은 수준이지만 제대로 된 풀이 방법을 갖지 않고 접근하면 어려운 문제였습니다. 저 역시 헤매다가 다른 분들의 풀이를 참고하며 문제를 이해했고 풀었습니다.
'Tests > 프로그래머스' 카테고리의 다른 글
[Programmers] 프로그래머스 파이썬 > 순위 (0) | 2020.10.06 |
---|---|
[Programmers] 프로그래머스 파이썬 > 여행경로 (0) | 2020.09.30 |
[Programmers] 프로그래머스 파이썬 > 입국심사 (0) | 2020.09.27 |
[Programmers] 프로그래머스 파이썬 > 단속카메라 (0) | 2020.09.26 |
[Programmers] 프로그래머스 파이썬 > 2 x n 타일링 (0) | 2020.09.24 |
댓글