부캠하느라 알고리즘은 오랜만에 했다..ㅎㅎ 그래도 DP는 놓지 않을 것..

70. Climbing Stairs

description

You are climbing a stair case. It takes n steps to reach to the top.

Each time you can either climb 1 or 2 steps. In how many distinct ways can you climb to the top?

Example 1:

Input: 2
Output: 2
Explanation: There are two ways to climb to the top.
1. 1 step + 1 step
2. 2 steps

Example 2:

Input: 3
Output: 3
Explanation: There are three ways to climb to the top.
1. 1 step + 1 step + 1 step
2. 1 step + 2 steps
3. 2 steps + 1 step

Constraints:

1 <= n <= 45

Solution

이번에는 2가지 버전으로 코드를 짜봤다.

Code 1

class Solution:
    def climbStairs(self, n: int) -> int:
        if (n == 1) : return 1
        elif (n == 2) : return 2
        
        stair = [1] * n
        stair[1] = 2
        
        for i in range(2,n) :
            stair[i] = (stair[i-1] + stair[i-2])
        
        return stair[n-1]
  • 전형적인 DP 문제처럼 배열을 사용했다.
  • n-1도 n-2에서 1 step을 간 것이므로 추가적으로 더해주는 것 없이 n-1과 n-2들을 더해서 n에 넣는다.

Code 2

class Solution:
    def climbStairs(self, n: int) -> int:
        if (n == 1) : return 1
        elif (n == 2) : return 2
        
        stair0 = 1
        stair1 = 2
        
        for i in range(2,n) :
            temp = stair1
            stair1 = stair0 + stair1
            stair0 = temp
        
        return stair1
  • 첫 번째 코드와 논리는 같다

코드 별 결과

ver Runtime Memory
1 24ms 14MB
2 32ms 13.8MB

뭐가 더 좋다고 할 수 있을까? 🤔

문제 출처

https://leetcode.com/problems/climbing-stairs/