BJ_S3_9461_파도반수열 - Java

[문제링크]

https://www.acmicpc.net/problem/9461

9461번: 파도반 수열

---

[문제]

오른쪽 그림과 같이 삼각형이 나선 모양으로 놓여져 있다. 첫 삼각형은 정삼각형으로 변의 길이는 1이다. 그 다음에는 다음과 같은 과정으로 정삼각형을 계속 추가한다. 나선에서 가장 긴 변의 길이를 k라 했을 때, 그 변에 길이가 k인 정삼각형을 추가한다.

파도반 수열 P(N)은 나선에 있는 정삼각형의 변의 길이이다. P(1)부터 P(10)까지 첫 10개 숫자는 1, 1, 1, 2, 2, 3, 4, 5, 7, 9이다.

N이 주어졌을 때, P(N)을 구하는 프로그램을 작성하시오.

---

[입력]

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, N이 주어진다. (1 ≤ N ≤ 100)

---

[출력]

각 테스트 케이스마다 P(N)을 출력한다.

---

[예제 입력 1]

2
6
12

---

[예제 출력 1]

3
16

---

[설명]

이 문제는 DP 문제이다.
문제의 그림을 잘 살펴보면 4번째 삼각형의 변의 길이가 1번째 삼각형의 변의 길이와 2번째 삼각형의 변의 길이의 합과 같다.
그러므로 DP[i] = DP[i-2] + DP[i-3]이 된다.
중요한 것은 숫자가 커지게 되면 int 형의 범위를 넘어가게 되므로 DP를 Long으로 선언해줘야 한다.

 

백준 알고리즘 9461번 JAVA풀이

 

import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner scann = new Scanner(System.in);
		int T=scann.nextInt();
		for(int t=0;t<T;t++) {
			int N=scann.nextInt();
			long[] dp = new long[N+1];
			dp[0]=0;
			dp[1]=1;
			if(N>1) {
				dp[2]=1;
			}
			for(int i=3;i<=N;i++) {
				dp[i]= dp[i-2]+dp[i-3];
			}
			System.out.println(dp[N]);
		}
	}
}

 

 

 

궁금하신 부분이나 부족한 점은 댓글로 알려주시면 감사하겠습니다.