-
[백준/Python] 2775번: 부녀회장이 될테야 풀이Coding Test/Baekjoon 2021. 5. 13. 20:52
출처: https://www.acmicpc.net/problem/2775
문제
평소 반상회에 참석하는 것을 좋아하는 주희는 이번 기회에 부녀회장이 되고 싶어 각 층의 사람들을 불러 모아 반상회를 주최하려고 한다.
이 아파트에 거주를 하려면 조건이 있는데, “a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다” 는 계약 조항을 꼭 지키고 들어와야 한다.
아파트에 비어있는 집은 없고 모든 거주민들이 이 계약 조건을 지키고 왔다고 가정했을 때, 주어지는 양의 정수 k와 n에 대해 k층에 n호에는 몇 명이 살고 있는지 출력하라. 단, 아파트에는 0층부터 있고 각층에는 1호부터 있으며, 0층의 i호에는 i명이 산다.
입력
첫 번째 줄에 Test case의 수 T가 주어진다. 그리고 각각의 케이스마다 입력으로 첫 번째 줄에 정수 k, 두 번째 줄에 정수 n이 주어진다
출력
각각의 Test case에 대해서 해당 집에 거주민 수를 출력하라.
제한
- 1 ≤ k, n ≤ 14
해결 방법
1. 0층부터 (k-1)층까지 각 층마다 0호부터 n호까지 계산한다.(문제는 1부터 n호까지이지만 코딩 편의상 0호부터 시작한다. 0층 0호는 0명으로 계산하면 더하여도 값은 그대로다.)
2. k층의 n호를 계산하여 출력한다.
팁
아파트를 [[1층],[2층],[3층],...]과 같이 2차원 배열에 저장해놓고 적절히 슬라이스하면 쉽게 구할 수 있다.
예를 들어, 배열에 (i-1)층까지 저장되어 있을 때 i층은 [sum(apt[i-1][:x+1] for x in range(n+1)]로 나타낼 수 있다. 위의 식을 풀어서 설명하면 다음과 같다.
- for x in range(n+1): x는 0부터 n까지 1씩 증가
- sum(apt[i-1][:x+1]: x호 사람 수 = (i-1)층 아파트의 0호부터 x호까지([:x+1])의 합
코드
'Coding Test > Baekjoon' 카테고리의 다른 글
[백준/Python] 16554번 소인수분해 풀이 (0) 2021.05.15 [백준/c++] 10757번: 큰 수 A+B (0) 2021.05.13 [백준/Python] 1929번: 소수 구하기 풀이 (0) 2021.05.13 [백준/Python] 4153번: 직각삼각형 풀이 (0) 2021.05.13 [백준/Python] 1085번: 직사각형에서 탈출 풀이 (0) 2021.05.13