[Python Algorithm] 완전탐색(DFS) - 라이브러리를 이용한 조합

> 완전탐색 - 백트랙킹, 상태트리와 CUT EDGE / DFS 기초 

https://sanamode.tistory.com/53 

[Python Algorithm] 완전탐색(DFS) - 수들의 조합

위의 버전에서 라이브러리를 이용해서 다시 !

⭐️ 라이브러리를 사용하지 말라는 곳도 다수 ! 알기만 하기 ⭐️

 

 

📖 문제 : 수들의 조합

N개의 정수가 주어지면 그 숫자들 중 K개를 뽑는 조합의 합이 임의의 정수 M의 배수인 개수는 몇 개가 있는지 출력하는 프로그램을 작성하세요.
예를 들면 5개의 숫자 2 4 5 8 12가 주어지고, 3개를 뽑은 조합의 합이 6의 배수인 조합을 찾으면 4+8+12 2+4+12로 2가지가 있습니다.

 

 

입력설명
첫줄에 정수의 개수 N(3<=N<=20)과 임의의 정수 K(2<=K<=N)가 주어지고, 두 번째 줄에는 N개의 정수가 주어진다.
세 번째 줄에 M이 주어집니다.

 

출력설명
총 가지수를 출력합니다.

 

입력예제 1

5 3
2 4 5 8 12
6

 

출력예제 1

2

import sys
import itertools as it

n, k = map(int, input().split())
a = list(map(int, input().split()))
m = int(input())
cnt = 0

for x in it.combinations(a, k):
	if sum(x)%m==0:
    	cnt+=1
        
print(cnt)

 

 

⭐️ Point ! ⭐️

• 라이브러리 itertools 사용 !
  
  
• 라이브러리 itertools를 통한 조합 
  for x in it.combinations(a, 3):
  print(x) 하면
  (2, 4, 5)
  (2, 4, 8) ....
  5C3 = 10개 출력

 

출처

• 인프런 : 파이썬 알고리즘 문제 풀이