[Java] BOJ 2775 부녀회장이 될테야

문제 

문제 링크 https://www.acmicpc.net/problem/2775

• 이 아파트에 거주를 하려면 조건이 있는데, “a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다” 는 계약 조항을 꼭 지키고 들어와야 한다.
• 아파트에 비어있는 집은 없고 모든 거주민들이 이 계약 조건을 지키고 왔다고 가정했을 때, 주어지는 양의 정수 k와 n에 대해 k층에 n호에는 몇 명이 살고 있는지 출력하라.
• 단, 아파트에는 0층부터 있고 각층에는 1호부터 있으며, 0층의 i호에는 i명이 산다.
• 첫 번째 줄에 Test case의 수 T가 주어진다. 그리고 각각의 케이스마다 입력으로 첫 번째 줄에 정수 k, 두 번째 줄에 정수 n이 주어진다.
• 각각의 Test case에 대해서 해당 집에 거주민 수를 출력하라.

아이디어

• 이 문제를 보고 어떻게 풀어야 되지 고민했었다... 나는 배열 두개를 두고 풀어야겠다고 생각했다.
• 첫번째 0층의 호수의 인원들은 정해져 있으므로 초기값은 정해놓고 시작을 했다.
• 두 배열은 현재층과 전층의 인원수를 저장하는데 사용했다.
• 반복문으로 1층부터 계산하였으며 한 층을 계산하고 나서 다음층으로 또 가야될 경우, 현재층의 인원 값을 전층의 배열의 호수에 맞게 다시 저장하고 현재층의 배열에 대해서는 1호는 계속 1명이므로 1로 초기화해주고 나머지는 0으로 초기화해준다. 
• 호수는 구해야 되는 호수 n호까지만 구하면 되므로 n만큼 반복해서 구했다. 

겪은 시행착오

• 처음에 알고리즘이 생각나지 않아 조금 헤맸었다.. (여담이지만 저런 계약조건이 있으면 누가 저 아파트에 들어가고 싶을까란 생각이 들었다...ㅎㅎ)

저 아파트는 그럼 고층으로 갈수록 인원이 많아야되니까 최고층 마지막 호수에는 슈퍼인싸가 살고 있지 않을까? ㅎㅎ

코드

import java.io.*;
import java.util.*;

public class BOJ2775 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int T = Integer.parseInt(br.readLine());

        for(int i = 0; i < T; i++){
            int k = Integer.parseInt(br.readLine());
            int n = Integer.parseInt(br.readLine());

            // 전층 인원수를 저장할 배열
            int apt[] = new int[n + 1];
            // 현재층 인원수를 저장할 배열
            int apt2[] = new int[n + 1];

            // 1호는 무조건 인원이 1명이므로 1로 고정
            apt2[1] = 1;

            for(int j = 1; j <= n; j++){
                apt[j] = j;
            }

            // 해당 층이 될때마다 반복해야되므로
            for(int j = 1; j <= k; j++){
                // 한 층에 호수가 여러개이므로 호마다 계산
                for(int l = 2; l <= n; l++){
                    // 한 호수에 대해 아래층 인원수로 계산
                    for(int m = 1; m <= l; m++){
                        // 현재 호수의 인원수에 구해야되는 호수까지의 전층 인원들을 더해줌
                        apt2[l] = apt2[l] + apt[m];
                    }
                }

                // 층이 마지막 층이라면 종료
                if(j == k) {
                  break;
                }
                // 마지막 층이 아닐 경우
                else{
                    for(int l = 1; l <= n; l++){
                        // 전 층의 인원수들을 현재층 인원으로 초기화
                        apt[l] = apt2[l];
                        // 현재층 인원수 배열을 초기화
                        if(l == 1){
                            // 1층은 인원이 1명이므로 1로 초기화
                            apt2[l] = 1;
                        }
                        else{
                            // 나머지는 0으로 초기화
                            apt2[l] = 0;
                        }
                    }
                }
            }

            System.out.println(apt2[n]);
        }
    }
}