Search

Find Common Elements Between Two Arrays

문제 설명 : 문제에서 제공한 answer1, answer2에 대한 답을 반환한다.
풀이 방법
nums 에 대한 숫자들의 갯수를 미리 dict로 만들어 저장한다.
이를 사용하여 answer를 구한다.
시간복잡도 : O(N)O(N)
성공 코드
from collections import defaultdict # O(N) class Solution: def findIntersectionValues(self, nums1: List[int], nums2: List[int]) -> List[int]: dict_for_answer1 = defaultdict(int) dict_for_answer2 = defaultdict(int) answer1 = 0 answer2 = 0 for num in nums1: dict_for_answer1[num] += 1 for num in nums2: dict_for_answer2[num] += 1 for num in set(nums2): answer1 += dict_for_answer1[num] for num in set(nums1): answer2 += dict_for_answer2[num] return [answer1, answer2]
Python
복사