[백준] 2775번 : 부녀회장이 될테야 - C++

문제

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

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

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

#include <iostream>
using namespace std;
int(* hwMny(void))[14] {
	static int apt[15][14] = {};
	for (int i = 0; i < 14; i++) {
		apt[0][i] = i + 1;
	}//1층에 사는 사람 수 저장
	for (int i = 1; i < 15; i++) {
		for (int j = 0; j < 14; j++) {
			int num = 0;
			for (int k = 0; k <= j; k++) {
				num += apt[i - 1][k];
			}
			apt[i][j] = num;
		}
	}
	return apt;
}
int main(void) {
	int(* ppleLiv)[14];
	ppleLiv = hwMny();
	int num;
	cin >> num;
	int k; int n;
	for (int i = 0; i < num; i++) {
		cin >> k >> n;
		cout << ppleLiv[k][n-1]<<endl;
	}
}

hwMny함수는 배열에 각 층별로, 호수별로 몇명씩 사는지 저장한다. 이 함수는 2차원 배열을 반환한다. 

int(* hwMny(void))[14]{...} 

이 함수는 14개의 요소를 가진 배열의 주소를 저장할 수 있는 포인터(배열포인터)를 반환한다.

hwMny함수 안에서 배열 apt는 static으로 선언해 주었다. (지역변수이기 때문에 함수가 끝나면 사라진다)

 

main함수에서 (*ppleLive)[14] 배열포인터에 hwMny()함수를 호출해 저장하고, 입력받은 층의 호수에 사는 인원을 출력한다.

 

//입력은 1부터 14까지 가능하지만 아파트에는 0층도 존재하기 때문에 배열의  row는 15여야함.