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

문제

자연수가 중복 없이 들어있는 배열이 있습니다. 이 배열에서 합이 K의 배수가 되도록 서로 다른 숫자 세개를 고르는 방법은 몇 가지인지 세려고 합니다.

자연수가 들어있는 배열 arr와 arr의 길이 arr_len이 매개변수로 주어질 때, 이 배열에서 합이 K의 배수가 되도록 서로 다른 숫자 세개를 고르는 방법의 가짓수를 return 하도록 solution 함수를 완성해주세요.

 

 

 

 

코드

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, "입니다.")