itertools > Maximize It!

Maximize It! | HackerRank

 

문제

---

 

 

=> 각 N개의 원소를 가지는 리스트 K개가 있다. 각 리스트에서 하나의 원소를 제곱하고 서로 더한 뒤 M으로 나눈 나머지를 S라고 하자. 얻을 수 있는 S의 최댓값을 구하라.

 

 

 

 

코드

---

from itertools import product
from itertools import accumulate


K, M = map(int, input().split())

li = []
for i in range(K):
    li.append(list(input().split()[1:]))
    

all_combi = list(product(*li))
maxs = []

for j in all_combi:
    j = list(map(int, j))

    maxs.append(list(accumulate(list(x**2 for x in j)))[-1])
    
print(
    max(list(map(lambda x: x%M, maxs)))
    )

 

 

 

 

노트

---

step1 가능한 모든 조합 추출

K, M = map(int, input().split())

li = []
for i in range(K):
    li.append(list(input().split()[1:]))
    

all_combi = list(product(*li))

• for반복문으로 각 리스트 받아오기(N값이 제일 먼저 input되기때문에 리스트에서 제외해줌
• product연산으로 iterable들의 모든 조합 뽑아내기

 

 

step2 제곱합들 구하기

for j in all_combi:
    j = list(map(int, j))

    maxs.append(list(accumulate(list(x**2 for x in j)))[-1])

• accumulate연산으로 누적합 구하기(마지막 누적합이 모든 제곱들의 합)
• list comprehension 유용하게

 

 

step3 나머지중 최댓값 구하기

print(
    max(list(map(lambda x: x%M, maxs)))
    )

• lambda 함수로 M으로 나눈 나머지들을 리스트에 담아냄

 

 

 

 

참조

---

변수 이름 for문으로 할당하는 방법(globals( ))

[python] 파이썬 f-string (문자열 포매팅 방법 3)

024 연간 매출액을 계산하려면? ― itertools.accumulate

3.5 람다(lambda)