[TIL] 자바 알고리즘 연습 240812 22일차

문제구분 : DP

문제 : https://school.programmers.co.kr/learn/courses/30/lessons/12914

프로그래머스

 

효진이는 멀리 뛰기를 연습하고 있습니다. 효진이는 한번에 1칸, 또는 2칸을 뛸 수 있습니다. 칸이 총 4개 있을 때, 효진이는
(1칸, 1칸, 1칸, 1칸)
(1칸, 2칸, 1칸)
(1칸, 1칸, 2칸)
(2칸, 1칸, 1칸)
(2칸, 2칸)
의 5가지 방법으로 맨 끝 칸에 도달할 수 있습니다. 멀리뛰기에 사용될 칸의 수 n이 주어질 때, 효진이가 끝에 도달하는 방법이 몇 가지인지 알아내, 여기에 1234567를 나눈 나머지를 리턴하는 함수, solution을 완성하세요. 예를 들어 4가 입력된다면, 5를 return하면 됩니다.

제한 사항
n은 1 이상, 2000 이하인 정수입니다.

 

풀이 :

언뜻 봐서는 피보나치 수열 문제와 크게 다른게 없어 보여 그대로 풀어 봤다.

class Solution {
    public long solution(int n) {
        long[] stairs = new long[n+1];
        stairs[1] = 1;
        stairs[2] = 2;
        for (int i = 3; i <= n; i++) {
            stairs[i] = (stairs[i-1] + stairs[i-2]) % 1234567;
        }
        
        return stairs[n];
    }
}

 

그러나 이번에도 쉽게 넘어가지 않고

테스트케이스 1번에서 런타임 에러가 발생함.

찾아보니 메모리 부족으로 인한 에러라고 한다.

 

 

배열에 저장하는 것을 포기했더니 한번에 성공

class Solution {
  public long solution(int n) {
      long answer = 1;
      long a=1,b=2;
      
      if(n==1) return 1;
      if(n==2) return 2;
      
      for(int i=3; i<=n; i++){
          answer=(a+b)%1234567;
          a=b;
          b=answer;
      }
      return answer;
  }
}