2309: 일곱 난쟁이

canyi·2023년 5월 6일
0

백준

목록 보기
4/19

문제링크

combinations

Python에서 "combinations" 함수는 "itertools" 모듈에서 제공하는 메서드로, 주어진 iterable 객체 (예: 리스트나 문자열)에서 주어진 길이의 모든 가능한 조합을 반환합니다.

  • "itertools.combinations" 메서드는 두 개의 인자를 받습니다: iterable 객체 (이 예제에서는 "my_list")와 생성하려는 조합의 길이 입니다.
  • 메서드는 이터레이터를 반환하므로, "list" 함수를 사용하여 리스트로 변환해주어야 합니다.
  • 결과 리스트에는 원래 리스트에서 길이가 2인 모든 가능한 조합이 포함됩니다.
from itertools import combinations

input_num = [int(input()) for _ in range(9)]

combi_sort = []
for combi in combinations(input_num, 7):
    if sum(combi) == 100:
        combi_sort = sorted(combi)

        for i in combi_sort:
            print(i)

함수 사용

input_num = [int(input()) for _ in range(9)]
input_num.sort()
tot_num = sum(input_num)


def f():
    for i in range(8):
        for j in range(i+1,9):
            if tot_num - input_num[i] - input_num[j] == 100:
                for k in range(9):
                    if i != k and j != k:
                        print(input_num[k])
                return


f()

profile
백엔드 개발 정리

0개의 댓글