백준 기본수학 1 풀이 - 4

백준 단계별로 풀어보기 - 기본수학1 문제 2개 python으로 풀기

---

2775 - 부녀회장이 될테야

※ 평소 반상회에 참석하는 것을 좋아하는 주희는 이번 기회에 부녀회장이 되고 싶어 각 층의 사람들을 불러 모아 반상회를 주최하려고 한다. 이 아파트에 거주를 하려면 조건이 있는데, “a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다” 는 계약 조항을 꼭 지키고 들어와야 한다. 아파트에 비어있는 집은 없고 모든 거주민들이 이 계약 조건을 지키고 왔다고 가정했을 때, 주어지는 양의 정수 k와 n에 대해 k층에 n호에는 몇 명이 살고 있는지 출력하라. 단, 아파트에는 0층부터 있고 각층에는 1호부터 있으며, 0층의 i호에는 i명이 산다.

T = int(input())
for _ in range(T):
  k = int(input()); n = int(input())
  floor = []
  for i in range(n):
    floor.append(i+1)
  for _ in range(k):
    for j in range(n-1):
      floor[j+1] += floor[j]
  print(floor[-1])

풀이 : 'a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다' 라는 조건을 보기 쉽게 표로 그려보면

4층	1	6	21	56
3층	1	5	15	35
2층	1	4	10	20
1층	1	3	6	10
0층	1	2	3	4

와 같이 그려볼 수 있다.

 k와 n을 입력받은 뒤 n만큼 0층에 인원을 채운다. 그 뒤 1호실을 제외하고(1호실은 모든 층이 1명이므로) 2호실 부터 시작하여 n호실의 인원 수를 계산하여 값을 교체해준다. 위 과정을 k번 반복하여 최종적으로 나온 floor 리스트의 맨 오른쪽 결과를 출력하여준다.

---

2839 - 설탕 배달

※ 상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다. 상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다. 상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.

N = int(input())
num_list = []; num = 0
for i in range(N // 5 + 1):
  if (N - 5*i) % 3 == 0: num = i + (N - 5*i) // 3; num_list.append(num)
if num_list == []: print(-1)
else: print(min(num_list))

풀이 : 가장 적은 수의 봉지를 배달하는 것이 목표이므로 해당 목표를 달성하기 위해서는 5kg짜리 봉지를 최대한으로 챙기는 것이 해답이라고 생각한다. 5kg 봉지를 0개부터 시작하여 한개씩 챙겼을 때 남은 목표량이 3으로 나누어 떨어지면 총 봉지수를 리스트에 저장하는 과정을 반복한다. 그리고나서 리스트내에 저장된 경우의 수들 중에서 가장 작은 수를 출력하여준다.