Koala - 15기/기초 알고리즘 스터디

[백준/python] 1350번 : 진짜 공간

lhy1839 2024. 7. 14. 14:49

 

풀이

def calculate_disk_space(files, cluster_size):
    total_space = 0
    for file_size in files:
        if file_size == 0:
            continue
        # 클러스터 수 계산
        clusters_needed = (file_size + cluster_size - 1) // cluster_size
        # 사용한 디스크 공간 계산
        total_space += clusters_needed * cluster_size
    return total_space

# 입력 받기
import sys
input = sys.stdin.read
data = input().strip().split()

# 첫 번째 줄: 파일의 개수 N
N = int(data[0])

# 두 번째 줄: 파일의 크기
file_sizes = list(map(int, data[1:N+1]))

# 세 번째 줄: 클러스터의 크기
cluster_size = int(data[N+1])

# 사용한 디스크 공간 계산 및 출력
print(calculate_disk_space(file_sizes, cluster_size))