sangjun1389님의 게시글

질문&답변

two_sum dictionary 적용 관련 질문드립니다.

안녕하세요. 같은 고민했던 기억이 있어 풀이 남겨두고 갑니다. 물론 저보다 빠르게 해결되는 풀이도 있지만, 딕셔너리로 간단하게 해결하는 방법이라도 참고하세요! 저는 인덱스로 value로 지정하여 담지 않고, 특정 원소가 있는 개수를 value로 사용하였습니다. 그리고 if 문을 추가하여 중복된 원소인지를 추가하는 로직을 통해 중복 키 여부를 판단하였습니당. 아래는 코드입니다. class Solution: def twoSum(self, nums: List[int], target: int) -> List[int]: _dict = {} results = [] for number in nums: if number in _dict : _dict[number] += 1 else : _dict[number] = 1 for index, number in enumerate(nums): if target - number in _dict: if target == number * 2 : if _dict[number] >= 2 : results.append(index) else : results.append(index) return results