[λ°±μ€€/python] 2798번 : λΈ”λž™μž­

2021. 4. 13. 21:41

문제

μΉ΄μ§€λ…Έμ—μ„œ 제일 인기 μžˆλŠ” κ²Œμž„ λΈ”λž™μž­μ˜ κ·œμΉ™μ€ μƒλ‹Ήνžˆ 쉽닀. μΉ΄λ“œμ˜ 합이 21을 λ„˜μ§€ μ•ŠλŠ” ν•œλ„ λ‚΄μ—μ„œ, μΉ΄λ“œμ˜ 합을 μ΅œλŒ€ν•œ 크게 λ§Œλ“œλŠ” κ²Œμž„μ΄λ‹€. λΈ”λž™μž­μ€ μΉ΄μ§€λ…Έλ§ˆλ‹€ λ‹€μ–‘ν•œ κ·œμ •μ΄ μžˆλ‹€.

ν•œκ΅­ 졜고의 λΈ”λž™μž­ 고수 김정인은 μƒˆλ‘œμš΄ λΈ”λž™μž­ κ·œμΉ™μ„ λ§Œλ“€μ–΄ 상근, μ°½μ˜μ΄μ™€ κ²Œμž„ν•˜λ €κ³  ν•œλ‹€.

김정인 λ²„μ „μ˜ λΈ”λž™μž­μ—μ„œ 각 μΉ΄λ“œμ—λŠ” μ–‘μ˜ μ •μˆ˜κ°€ μ“°μ—¬ μžˆλ‹€. κ·Έ λ‹€μŒ, λ”œλŸ¬λŠ” Nμž₯의 μΉ΄λ“œλ₯Ό λͺ¨λ‘ μˆ«μžκ°€ 보이도둝 λ°”λ‹₯에 λ†“λŠ”λ‹€. 그런 후에 λ”œλŸ¬λŠ” 숫자 M을 크게 μ™ΈμΉœλ‹€.

이제 ν”Œλ ˆμ΄μ–΄λŠ” μ œν•œλœ μ‹œκ°„ μ•ˆμ— Nμž₯의 μΉ΄λ“œ μ€‘μ—μ„œ 3μž₯의 μΉ΄λ“œλ₯Ό 골라야 ν•œλ‹€. λΈ”λž™μž­ λ³€ν˜• κ²Œμž„μ΄κΈ° λ•Œλ¬Έμ—, ν”Œλ ˆμ΄μ–΄κ°€ κ³ λ₯Έ μΉ΄λ“œμ˜ 합은 M을 λ„˜μ§€ μ•ŠμœΌλ©΄μ„œ Mκ³Ό μ΅œλŒ€ν•œ κ°€κΉκ²Œ λ§Œλ“€μ–΄μ•Ό ν•œλ‹€.

Nμž₯의 μΉ΄λ“œμ— 써져 μžˆλŠ” μˆ«μžκ°€ μ£Όμ–΄μ‘Œμ„ λ•Œ, M을 λ„˜μ§€ μ•ŠμœΌλ©΄μ„œ M에 μ΅œλŒ€ν•œ κ°€κΉŒμš΄ μΉ΄λ“œ 3μž₯의 합을 ꡬ해 좜λ ₯ν•˜μ‹œμ˜€.

 

μž…λ ₯

첫째 쀄에 μΉ΄λ“œμ˜ 개수 N(3 ≤ N ≤ 100)κ³Ό M(10 ≤ M ≤ 300,000)이 주어진닀. λ‘˜μ§Έ μ€„μ—λŠ” μΉ΄λ“œμ— μ“°μ—¬ μžˆλŠ” μˆ˜κ°€ 주어지며, 이 값은 100,000을 λ„˜μ§€ μ•ŠλŠ” μ–‘μ˜ μ •μˆ˜μ΄λ‹€.

합이 M을 λ„˜μ§€ μ•ŠλŠ” μΉ΄λ“œ 3μž₯을 찾을 수 μžˆλŠ” 경우만 μž…λ ₯으둜 주어진닀.

 

좜λ ₯

첫째 쀄에 M을 λ„˜μ§€ μ•ŠμœΌλ©΄μ„œ M에 μ΅œλŒ€ν•œ κ°€κΉŒμš΄ μΉ΄λ“œ 3μž₯의 합을 좜λ ₯ν•œλ‹€.

 

μ •λ‹΅

from itertools import combinations
import sys
input = sys.stdin.readline

N, M = map(int, input().split())
cards = list(map(int, input().split()))

answer = []
for i in combinations(cards, 3):
    if sum(i) <= M:
        answer.append(sum(i))

print(max(answer))

combinationsλ₯Ό μ΄μš©ν•˜μ—¬ λͺ¨λ“  경우의 수λ₯Ό ꡬ해 answer에 μ €μž₯ν•˜κ³  κ·Έ 쀑 κ°€μž₯ 큰 수λ₯Ό 좜λ ₯ν•œλ‹€.

 

μ•„λž˜λŠ” combinations에 λŒ€ν•œ ν¬μŠ€νŒ…μ΄λ‹€.

2021.04.01 - [πŸ”μ•Œκ³ λ¦¬μ¦˜/πŸ”…κ°œλ…μ •λ¦¬] - [파이썬/python] μˆœμ—΄κ³Ό μ‘°ν•© : itertools μ΄μš©ν•˜κΈ° (permutations, combinations)

 

[파이썬/python] μˆœμ—΄κ³Ό μ‘°ν•© : itertools μ΄μš©ν•˜κΈ° (permutations, combinations)

itertoolsλ₯Ό μ΄μš©ν•˜μ—¬ μˆœμ—΄κ³Ό 쑰합을 κ΅¬ν˜„ν•˜λ©΄ λ”μš± λΉ λ₯΄κ²Œ μ²˜λ¦¬ν•  수 μžˆλ‹€. 1. μˆœμ—΄ : permutations(N, r) Nμ—μ„œ 쀑볡을 ν—ˆμš©ν•˜μ§€ μ•Šκ³ , r개λ₯Ό 뽑아 μˆœμ„œλŒ€λ‘œ λ‚˜μ—΄ν•œλ‹€. 쀑볡을 ν—ˆμš©ν•˜μ§€ μ•ŠκΈ° λ•Œλ¬Έμ—, 뽑

ye333.tistory.com

 

728x90

BELATED ARTICLES

more