[COS Pro 1급, Python] 2차 4번 : 합이 k 배가 되는 수

문제 유형

 코딩

난이도 (주관적인)

 hard

Note 

 1. Combination(조합).

i, j, k

 배열의 원소들 중 3개를 택하여 더하여 K배수가 되는지 확인. 

 

Code

# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean
# 2차 4번

def solution(arr, K):
	answer = 0
	
	for i in range(len(arr)-2):
		for j in range(i+1, len(arr)-1):
			for k in range(j+1, len(arr)):
				if (arr[i]+arr[j]+arr[k])%K == 0:
					answer += 1
	
	return answer

arr = [1, 2, 3, 4, 5]
K = 3
ret = solution(arr, K)

print("solution 함수의 반환 값은", ret, "입니다.")

 

※ 가끔 코드 중 print(~)가 있습니다. 정리 못한 점 죄송합니다.