[Python] 백준 16769번 Mixing Milk

 

 

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

16769번: Mixing Milk

 

| 문제 내용

| 풀이

import sys

input = sys.stdin.readline
# c, m 배열 받기
c, m = [0 for i in range(3)], [0 for i in range(3)]
for i in range(3):
    c[i], m[i] = map(int, input().split())

# 현재 인덱스 밀크를 다음 인덱스%3 병의 밀크로 옮길수 있는 최대치 옮기기(현재 병, 다음 병 밀크값 갱신)
for i in range(100):
    idx = i % 3
    next_idx = (idx + 1) % 3
    # 남길 양, 넘길 양 구하기
    possible_v = m[idx] + m[next_idx]
    remain_v, next_v = (0, possible_v) if possible_v <= c[next_idx] else (possible_v - c[next_idx], c[next_idx])
    # 갱신
    m[idx], m[next_idx] = remain_v, next_v

for num in m:
    print(num)

 

좀 더 깔끔한 코드

import sys

input = sys.stdin.readline
# c, m 배열 받기
c, m = [0 for i in range(3)], [0 for i in range(3)]
for i in range(3):
    c[i], m[i] = map(int, input().split())

# 현재 인덱스 밀크를 다음 인덱스%3 병의 밀크로 옮길수 있는 최대치 옮기기(현재 병, 다음 병 밀크값 갱신)
for i in range(100):
    idx = i % 3
    next_idx = (idx + 1) % 3
    # 현재, 다음 병 우유량 갱신
    m[idx], m[next_idx] = max(m[idx] - (c[next_idx] - m[next_idx]), 0), min(c[next_idx], m[idx] + m[next_idx])

for num in m:
    print(num)