https://www.acmicpc.net/problem/2775
1. 아이디어
다이나믹 프로그래밍으로 간단하게 구할 수 있다.
2. 문제풀이
아파트를 2차원 배열로 나타냈을 때 a층 b호에 살려면 a-1층의 1호부터 b호까지의 사람들의 수의 합만큼 사람들을 데려와야하는데 a층 b-1호에 살려면 a-1층의 1호부터 b-1호까지의 사람들을 데려와야 한다. 즉 a층 b호의 사람 수는 a층 b-1호 + a-1층 b호 임을 이용해서 점화식을 작성하는 방식으로 구현했다.
아파트의 거주민은 한번만 구하고 재사용해도 되는데 그냥 반복문 안에 깔끔하게 넣었다.
3. 코드
import java.io.*;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
StringBuilder sb = new StringBuilder();
int T = Integer.parseInt(br.readLine());
for (int tc = 1; tc <= T; tc++) {
int k = Integer.parseInt(br.readLine());
int n = Integer.parseInt(br.readLine());
int[][] dp = new int[1 + k][1 + n];
for (int i = 1; i <= n; i++) {
dp[0][i] = i;
}
for (int i = 1; i <= k; i++) {
for (int j = 1; j <= n; j++) {
dp[i][j] = dp[i - 1][j] + dp[i][j - 1];
}
}
sb.append(dp[k][n]).append("\n");
}
bw.write(sb.toString());
bw.flush();
}
}
4. 후기
'코딩테스트 준비 > 백준' 카테고리의 다른 글
[백준] 25305번 - 커트라인 [Java] (1) | 2025.01.08 |
---|---|
[백준] 25304번 - 영수증 [Java] (0) | 2025.01.08 |
[백준] 10162번 - 전자레인지 [Java] (0) | 2025.01.07 |
[백준] 1644번 - 소수의 연속합 [Java] (0) | 2025.01.07 |
[백준] 2960번 - 에라토스테네스의 체 [Java] (0) | 2025.01.07 |