You Nicholas

가고 가고 또 가면 알게되고, 행하고 행하고 또 행하게 되면 이루게 된다.

1 분 소요

1. 문제Permalink

[Bronze I] 부녀회장이 될테야 - 2775Permalink

문제 링크

성능 요약Permalink

메모리: 30860 KB, 시간: 80 ms

분류Permalink

구현(implementation), 수학(math)

문제 설명Permalink

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

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

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

입력Permalink

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

출력Permalink

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

출처: 백준, https://https://www.acmicpc.net/

2. 해결방법 시간복잡도Permalink

  1. 단순 코딩 O(N^3)

3. 문제 해결 및 코드Permalink

T = int(input()) # 테스트 케이스 T
for i in range(T):
k = int(input()) # 층수
n = int(input()) # 호수
floor = [i for i in range(1, n+1)] # 전체 층수의 사람들을 하나의 리스트로 만들기
for i in range(k): # k 층수만큼 반복
for j in range(1, n): # n 호수 + 1만큼 반복
floor[j] += floor[j-1] # 기존에 있던 리스트에 호수의 사람들을 계산한 만큼 넣기
print(floor[-1]) # 최종 목표인 (k-1)층의 사람수 세기
view raw 2775.py hosted with ❤ by GitHub

  • 주석을 참고하면서 이해를 돕습니다.Permalink

4. 알고리즘 및 해설Permalink

  1. 부녀회장 문제의 경우 각 층의 거주자를 알아야하는 문제이므로 층수와 호수를 모두 통틀어 전체 층수의 사람들을 하나의 리스트로 만들어준다.
  2. 이후 층수만큼의 반복문과 호수 만큼의 반복문을 통해 해당 층수 호수에 이전 호수의 사람들을 계산한 값을 넣어준다.
  3. 최종적으로 k-1층의 사람수를 출력한다.

참고