[파이썬] 백준 2775번 부녀회장이 될테야 문제 풀이 python

문제 설명

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

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

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

문제 링크

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

2775번: 부녀회장이 될테야

 

입출력 예시

프로그램 코드

li = [[1,2,3,4,5,6,7,8,9,10,11,12,13,14]]	#모든 층 모든 호수 값 저장할 2차원 리스트

for i in range(1, 15):	#모든 층 모든 호수 값 계산
    for j in range(14):
        if j == 0:
            li.append([1])
            continue
        li[i].append(li[i-1][j] + li[i][j-1]) 

T = int(input())	#T 입력
for i in range(T):	#입력받은 층 호수에 대한 인원 출력
    k = int(input())
    n = int(input())
    print(li[k][n-1])

 

프로그램 코드 설명

처음에는 f(k, n) = f(k-1, n) + f(k, n-1) 이런 식으로 재귀함수를 써서 문제를 해결하려고 했는데,시간 초과로 문제를 계속 틀려서 그냥 n이 14보다 작거나 같길래 모든 층에 있는 모든 호수 숫자를 다 구해줬습니다.만약 j == 0 이면 리스트에 [1] 을 추가하는데요이 코드는 새로운 층 리스트를 구할 때 새로운 리스트를 만들어 주기 위함입니다.그 후에는 f(k, n) = f(k-1, n) + f(k, n-1) 이기 때문에 값을 계산해서 리스트에 추가해줬습니다.

이렇게 모든 층의 모든 호수 숫자를 구한 후

입력받은 층과 호수에 따라서 2차원 리스트에 저장된 값을 출력해줬습니다.