알고리즘/Baekjoon
[CodingTest] python 백준 Baekjoon #2798 블랙잭 / 정렬
Happyoon ~
2021. 11. 20. 18:41
728x90
반응형
https://www.acmicpc.net/problem/2798
2798번: 블랙잭
첫째 줄에 카드의 개수 N(3 ≤ N ≤ 100)과 M(10 ≤ M ≤ 300,000)이 주어진다. 둘째 줄에는 카드에 쓰여 있는 수가 주어지며, 이 값은 100,000을 넘지 않는 양의 정수이다. 합이 M을 넘지 않는 카드 3장
www.acmicpc.net
나의 풀이
import sys
from itertools import combinations
#입력받은 값 저장 리스트
n,m = map(int,sys.stdin.readline().split())
arr = list(map(int,sys.stdin.readline().split()))
#조합은 튜플 반환.
#3개씩 숫자 조합하고 조합한 숫자의 합이 m보다 작을 때만 lst 리스트에 해당 조합 튜플 넣기
lst = [item for item in combinations(arr,3) if sum(item)<=m]
#튜플의 합에서 m을 빼주고, 뺀 값을 키로, 튜플을 값으로 하는 딕셔너리 생성
dic = {sum(item)-m:item for item in lst}
#튜플의 합과 m의 차가 가장 큰(즉, 0에 가까운) 값을 딕셔너리의 키로하는 튜플 값의 합 출력
print(sum(dic[max(dic)]))
조합과 딕셔너리를 활용했다.
다른 분들은 3중 포문을 사용했던데 그게 메모리도 적게 쓰고 더 시간이 잛게 걸리더라 ㅠ
메모리랑 시간복잡도 너무 어려움 .. ㅠ
반응형