[Python/파이썬] 백준 2845번 - 파티가 끝나고 난 뒤

문제

파티가 끝나고 나면, 사람들은 누가 파티에 왔는지와 얼마나 많은 사람들이 왔는지를 궁금해한다. 보통 파티는 매우 크게 열리기 때문에, 정확하게 몇 명이 참가했는지 알 수가 없다.

지난주 토요일에 상근이는 자신의 3학년 진학을 기념하면서 매우 성대한 파티를 열었다. 그리고, 상근이는 $1m^2$당 몇 명의 사람이 있었는지 알고있다.

상근이의 파티는 정말 엄청난 규모였기 때문에, 대부분의 신문에도 기사가 실렸다. 상근이는 서로 다른 5개의 신문을 보면서 그 기사에 적혀져있는 참가자의 수를 적었다.

상근이는 자신이 알고있는 참가자의 수가 정확하다고 생각한다. 각 신문 기사에 실려있는 참가자의 수가 몇 명 만큼 잘못되어있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 $1m^2$당 사람의 수 $L (1 ≤ L ≤ 10)$과 파티가 열렸던 곳의 넓이 $P (1 ≤ P ≤ 1000)$가 주어진다.

둘째 줄에는 각 기사에 실려있는 참가자의 수가 주어진다. 106보다 작은 양의 정수 5개가 주어진다.

출력

출력은 첫째 줄에 다섯 개의 숫자를 출력해야 한다. 이 숫자는 상근이가 계산한 참가자의 수와  각 기사에 적혀있는 참가자의 수의 차이이다.

풀이

• 실제 참가한 사람들을 입력받은 $L, P$로 계산하여 all_people로 받아준다.
• 뉴스에서 보도된 파티 참가 인원을 리스트 in_newspaper로 받아준다.
• for 반복문을 돌리며 in_newspaper의 원소들과 all_people과의 차이를 계산하여 wrong 리스트에 저장해준다.
• 몇 명 만큼 잘못되어 있는지 wrong 리스트의 원소들을 출력한다.

코드

l, p = map(int, input().split())
in_newspaper = list(map(int, input().split()))

all_people = l*p

wrong = []
a = 0

for i in range(5):
    a = in_newspaper[i] - all_people
    wrong.append(a)

print(*wrong)