미소를뿌리는감자의 코딩

[백준 2024/02/23] 11399번 ATM - python ver. 본문

코딩 테스트/백준

[백준 2024/02/23] 11399번 ATM - python ver.

미뿌감 2024. 2. 23. 21:47
728x90

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

 

11399번: ATM

첫째 줄에 사람의 수 N(1 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 각 사람이 돈을 인출하는데 걸리는 시간 Pi가 주어진다. (1 ≤ Pi ≤ 1,000)

www.acmicpc.net

 

1. 접근 방법

조금 직관적인 문제였다고 생각한다. 또한 문제에서 예시를 잘 들어주어서 이해하기가 좋았다.

 

딱히 추가적으로 설명할 부분이 없는 것 같다.

 

2. 코드

def sort_wait_time():
    n = int(input())
    withdrawal = list(map(int, input().split()))
    withdrawal.sort()
    return withdrawal


def total_wait_time(withdrawal):
    wait_time = 0
    sum_time = 0
    for i in range(len(withdrawal)):
        wait_time = wait_time + withdrawal[i]
        sum_time += wait_time

    return sum_time


def main():
    withdrawal = sort_wait_time()
    print(total_wait_time(withdrawal))


if __name__ == "__main__":
    main()
728x90