N-th Tribonacci Number (leetcode 1137)

The Tribonacci sequence Tn is defined as follows: 

T0 = 0, T1 = 1, T2 = 1, and Tn+3 = Tn + Tn+1 + Tn+2 for n >= 0.

Given n, return the value of Tn.

 

Example 1:

Input: n = 4 Output: 4 Explanation: T_3 = 0 + 1 + 1 = 2 T_4 = 1 + 1 + 2 = 4

Example 2:

Input: n = 25 Output: 1389537

 

Constraints:

• 0 <= n <= 37
• The answer is guaranteed to fit within a 32-bit integer, ie. answer <= 2^31 - 1.

트리보나치(n-2+n-1+n) 수를 구하는 문제이다. 주의사항으로는 n이 37까지이다.

 

본인의 경우 두가지 방법으로 풀어보았다.

첫번째 방법은 재귀 함수를 사용하지 않고 푼것이다.

 

class Solution {
    public int tribonacci(int n) {
        if (n < 2) {
        return n;
        }
        int a = 0, b = 1, c = 1, d;
//한칸씩 가면서 n까지 트리보나치를 구함
        while (n> 2) {
            d = a + b + c;
            a = b;
            b = c;
            c = d;
            n--;
        }
        return c;
    }
    
}

class Solution {
    public int tribonacci(int n) { if(n==2||n==1) {
      return 1;
    } 
                                  if(n==0) {
         return 0;   
            }else {
        return tribonacci(n-1)+tribonacci(n-2)+tribonacci(n-3);
        }
                                  
    }
}

그래서 배열을 이용해서 다시 풀어 보았다.

class Solution {
    static int[] arr = new int[38];
    public int tribonacci(int n) {
        if(n == 0) 
            return 0;
        if(n == 1 ||n == 2) 
            return 1;
        
        if(arr[n]!=0) return arr[n];
        arr[n] = tribonacci(n-3)+tribonacci(n-2)+tribonacci(n-1);
        return arr[n];
    }
    
}

배열에 있는 숫자의 경우 그냥 리턴하기 때문에 시간이 훨씬 줄어든다.