백준 2775 - 부녀회장이 될테야(파이썬)

문제

 

https://www.acmicpc.net/problem/2775

 

---

제출 코드

import sys
def plusFloor(preFloor):
    nextFloor = [sum(preFloor[:i + 1]) for i in range(len(preFloor))]
    return nextFloor

for _ in range(int(sys.stdin.readline().rstrip())):
    k = int(sys.stdin.readline().rstrip())
    n = int(sys.stdin.readline().rstrip())
    liveFloor = [i for i in range(1, n + 1)]
    for _ in range(k):
        liveFloor = plusFloor(liveFloor)
    print(liveFloor[-1])

코드 풀이 :

0층의 인원으로 자신이 사는 호까지만 만들어서 초기화를 하였고

다음층은 이전층의 자신이 사는 호까지의 합이므로 간단하게 함수로 구현 완료!!

 

효율성 때문에 이전 문제들과 같이 수학적 식을 찾아서 하려 했으나 n이 14까지로 제한되어있어 굳이 규칙을 찾아서 사용하지 않아도 효율성이 크게 달라지지 않을 것 같고 빨리 구현하는 게 목적이므로 그냥 빠르게 구현했다!