본문 바로가기

Algorithm/Python

060 - 동전 분배하기(DFS)

N개의 동전을 A, B, C 세명에게 나누어 주려고 합니다.
세 명에게 동전을 적절히 나누어 주어, 세 명이 받은 각각의 총액을 계산해, 총액이 가장 큰 사람과 가장 작은 사람의 차가 최소가 되도록 해보세요.
단 세 사람의 총액은 서로 달라야 합니다.

 

 

입력설명
첫째 줄에는 동전의 개수 N(3<=N<=12)이 주어집니다. 그 다음 N줄에 걸쳐 각 동전의 금액이 주어집니다.

 

출력설명

총액이 가장 큰 사람과 가장 작은 사람의 최소차를 출력하세요.

 

▣  입력예제

7
8
9

11

12

23

15

17

 

▣  출력예제

5

 

 

해설 : 29(12+17), 32(8+9+15), 34(11+23) 로 분배하면 최대금액과 최소금액의 차가 5가 되어 5가 최소차가 된다.

 

 

 

def DFS(L):
    global res

    if L == n:
        cha = max(money)-min(money)
        if money[0] != money[1] and money[0] != money[2] and money[1] != money[2]:
            if (cha < res):
                res = cha
                # print(f'res:{res}', end=' ')
                # for i in (x):
                #     print(f'x:{sum(i)}{i}', end=' ')
                # print()
        return
    else:
        for i in range(3):
            x[i].append(a[L])
            money[i] += a[L]
            DFS(L+1)
            money[i] -= a[L]
            x[i].pop()


if __name__ == '__main__':
    n = int(input())
    a = []
    res = float('inf')
    # list "x" is for debug
    x = [[] for _ in range(3)]

    money = [0]*3
    for i in range(n):
        a.append(int(input()))
    avg = sum(a)/3
    DFS(0)
    print(res)