알고리즘 단계별로 풀어보기 : BOJ-2775(부녀회장이될테야)

문제(출처)

평소 반상회에 참석하는 것을 좋아하는 주희는 이번 기회에 부녀회장이 되고 싶어 각 층의 사람들을 불러 모아 반상회를 주최하려고 한다.

이 아파트에 거주를 하려면 조건이 있는데, “a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다” 는 계약 조항을 꼭 지키고 들어와야 한다.

아파트에 비어있는 집은 없고 모든 거주민들이 이 계약 조건을 지키고 왔다고 가정했을 때, 주어지는 양의 정수 k와 n에 대해 k층에 n호에는 몇 명이 살고 있는지 출력하라. 

단, 아파트에는 0층부터 있고 각층에는 1호부터 있으며, 0층의 i호에는 i명이 산다.

입력

첫 번째 줄에 Test case의 수 T가 주어진다. 

그리고 각각의 케이스마다 입력으로 첫 번째 줄에 정수 k, 두 번째 줄에 정수 n이 주어진다. 

(1 <= k <= 14, 1 <= n <= 14)

출력

각각의 Test case에 대해서 해당 집에 거주민 수를 출력하라.

예제 입력

2

1

3

2

3

예제 출력

6

10

내 풀이

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.IOException;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

public class Main {

    public static void main(String args[]) {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        try {

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

            int result = 0;

            for (int i = 0; i < num; i++) {

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

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

                result = calc(num1, num2);

                bw.write(String.valueOf(result));

                bw.newLine();

            }

            bw.flush();

            bw.close();

        } catch (NumberFormatException e) {

            e.printStackTrace();

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    private static int calc(int num1, int num2) {

//      num1 : 층수

//      num2 : 호

//      2층  01  04  10  20 35  56  84  120 165 220

//      1층  01 03 06 10 15 21 28 36 45 55

//      0층  01 02 03 04 05 06 07 08 09 10

        int[][] apartment = new int[15][15];

        for (int j = 1; j <= apartment[0].length; j++) {

//          0층에 대한 초기화를 합니다. j-1로 0층 1호(0번째) 인덱스를 선택하여 j를 증분하여 저장합니다.

            apartment[0][j-1] = j;

        }

        if(num1 == 0 ) {

//          층수가 0층이면 바로 반환합니다.

            return num2;

        }

        

        for (int i = 1; i <= num1; i++) {

//          층수가 1층 이상부터 1씩 증분합니다.

            for (int j = 0; j < num2; j++) {

//              호수를 0부터 1씩 증분합니다.

//              이전 층 num2-1의 합을 저장 할 변수를 선언합니다.

                

//              해당 층수와 호수에 맞게 저장합니다.

                apartment[i][j] = sum(j,i,apartment);

            }

        }

        return apartment[num1][num2-1];

    }

    

    private static int sum(int j, int i, int[][] apartment) {

        int sum = 0;

        for(int k = 0; k <= j; k++) {

//          현재 층의 호수의 사람수를 저장하기 위해 이전 층의 사람수를 누적합니다.

            sum += apartment[i-1][k];

        }

        return sum;

    }

}

내 풀이 해석

층을 입력 받은 num1로 1차 반복하고 호를 입력 받은 num2로 2차 반복합니다.

sum 함수를 호출해서 해당 층-1의 해당 호까지 합을 더해 반환합니다.

반환 된 값을 이중 배열에 저장하고 반환합니다.

아쉬운 점

이중반복문 안에 들은 함수에서의 반복문.

의식의 흐름대로만 풀었습니다.

리팩토링이 필요하지 않나 생각하지만 통과도이서 넘어가기로 했습니다.