-
70128. 내적(Python, Programmers Level 1)Programmers 2022. 9. 4. 23:19
문제 설명
1차원 배열 a, b가 주어질 때, 동일한 위치의 요소를 서로 곱하고, 그 결과를 모두 더해 반환합니다.
코드 1.
a, b에서 각 요소를 차례대로 가져옵니다.
for i, j in zip(a, b):
그 수를 서로 곱해 결과에 더합니다.
answer += i * j
더보기def solution(a, b): answer = 0 for i, j in zip(a, b): answer += i * j return answer if __name__ == '__main__': result = solution([1, 2, 3, 4], [-3, -1, 0, 2]) print(3 == result, result) result = solution([-1, 0, 1], [1, 0, -1]) print(-2 == result, result)
코드 2.
a, b를 numpy array로 만듭니다.
a = array(a) b = array(b)
서로 곱하고, 더한 후 int로 변환하여 반환합니다.
return int(sum(a * b))
더보기from numpy import array def solution(a, b): a = array(a) b = array(b) return int(sum(a * b)) if __name__ == '__main__': result = solution([1, 2, 3, 4], [-3, -1, 0, 2]) print(3 == result, result) result = solution([-1, 0, 1], [1, 0, -1]) print(-2 == result, result)
'Programmers' 카테고리의 다른 글
1845. 폰켓몬(Python, Programmers Level 1) (0) 2022.09.05 12977. 소수 만들기(Python, Programmers Level 1) (0) 2022.09.05 76501. 음양 더하기(Python, Programmers Level 1) (0) 2022.09.04 86051. 없는 숫자 더하기(Python, Programmers Level 1) (0) 2022.09.04 64061. 크레인 인형뽑기 게임(Python, Programmers Level 1) (0) 2022.09.04