탑코더 알고리즘 정복 - 1. 키위 주스

class KiwiJuiceEasy:
	def thePouring(self, capacities, bottles, fromId, told):

# Case# 0
capacities = [20, 20]
bottles = [5, 8]
from_id = [0]
to_id = [1]

result : [0, 13]

# Case# 1
capacities = [10, 10]
bottles = [5, 8]
from_id = [0]
to_id = [1]

result : [3, 10]

# Case# 2
capacities = [30, 20, 10]
bottles = [10, 5, 5]
from_id = [0, 1, 2]
to_id = [1, 2, 0]

result : [10, 10, 0]

# Case# 3
capacities = [14, 35, 86, 58, 25, 62]
bottles = [6, 34, 27, 38, 9, 60]
from_id = [1, 2, 4, 5, 3, 3, 1, 0]
to_id = [0, 1, 2, 4, 2, 5, 3, 1]

result : [0, 14, 65, 35, 25, 35]

# Case# 4
capacities = [700000, 800000, 900000, 1000000]
bottles = [478478, 478478, 478478, 478478]
from_id = [2, 3, 2, 0, 1]
to_id = [0, 1, 1, 3, 2]

result : [0, 156956, 900000, 856956]

class KiwiJuiceEasy:
    def the_pouring(self, capacities, bottles, from_id, to_id):
        ret = []

        for i in range(len(from_id)):
            from_i = from_id[i]
            to_i = to_id[i]

            # 옮길 병이 비어있으면 pass
            if bottles[from_i] == 0:
                continue

            # 담길 병이 꽉차있으면 pass
            if bottles[to_i] == capacities[to_i]:
                continue

            # 옮길 병의 용량보다 from + to의 내용물이 적거나 같으면
            # bottles[to_i]에 주스를 옮김.
            if bottles[from_i] + bottles[to_i] <= capacities[to_i]:
                bottles[to_i] += bottles[from_i]
                bottles[from_i] = 0

            # 옮길 병의 용량보다 from + to의 내용물이 많으면
            # bottles[to_i]에 주스를 옮기고 나머지를 from에 남김.
            elif bottles[from_i] + bottles[to_i] > capacities[to_i]:
                tmp = bottles[to_i] + bottles[from_i] - capacities[to_i]
                bottles[to_i] = capacities[to_i]
                bottles[from_i] = tmp

        # 모든 bottles를 ret에 담음
        for bottle in bottles:
            ret.append(bottle)

        return ret

class KiwiJuiceEasy:
    def the_pouring(self, capacities, bottles, from_id, to_id):
        for i in range(len(from_id)):
            f = from_id[i]
            t = to_id[i]

            # 옮길 주스의 양과 기존 주스 병의 남은 용량을 비교해
            # 작은 것이 이동량이 된다.
            vol = min(bottles[f], capacities[t] - bottles[t])

            # 옮길 주스 병에 이동량을 뺀다.
            bottles[f] -= vol
            # 기존 주스 병의 남은 용량에 이동량을 더한다.
            bottles[t] += vol

        return bottles

class KiwiJuiceEasy:
    def the_pouring(self, capacities, bottles, from_id, to_id):
        for i in range(len(from_id)):
            sum = bottles[from_id[i]] + bottles[to_id[i]]
            bottles[to_id[i]] = min(sum, capacities[to_id[i]])
            bottles[from_id[i]] = sum - bottles[to_id[i]]

        return bottles