티스토리 뷰

1. 일단 모든 조합의 합을 구해서 B보다 같거나 큰것이 있다면 리스트에 넣어준다.

2. 리스트에서 가장 작은 값을 추출하여 B에서 뺀다.


from itertools import combinations

T = int(input())
#높이가 B 이상인 탑 중에서 높이가 가장 낮은 탑
#조합을 구하여 합이 B보다 큰것의 리스트를 만든다.
for tc in range(1,T+1):
    N, B = map(int,input().split()) # N:점원 수 B:선반의 높이
    H = list(map(int,input().split()))
    over = []
    sum_o = []
    for i in range(N+1):
        over.append(list(combinations(H,i)))
        for j in range(len(over[i])):
            if sum(over[i][j]) >= B:
                sum_o.append(sum(over[i][j]))
    
    print("#{} {}".format(tc, min(sum_o)-B))

'알고리즘 > SWEA' 카테고리의 다른 글

[SWEA] 5248. 그룹 나누기  (0) 2021.04.24
[SWEA] 5247. 연산  (0) 2021.04.23
[SWEA] 1970.쉬운 거스름돈  (0) 2021.04.16
[SWEA] 2819. 격자판의 숫자 이어붙이기  (0) 2021.04.16
[SWEA] 4366. 정식이의 은행업무  (0) 2021.04.16
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2024/11   »
1 2
3 4 5 6 7 8 9
10 11 12 13 14 15 16
17 18 19 20 21 22 23
24 25 26 27 28 29 30
글 보관함