📃 문제 설명

Check If N and Its Double exist

[문제 출처 : LeetCode]

👨‍💻 해결 방법

주어진 정수배열에서,
특정 원소 n에 대하여 n*2 원소가 있는지 체크하는 문제

처음에는 그냥 for와 in을 사용하여서 단순하게 체크했지만,
시간복잡도를 좀 더 개선해보고 싶었다.

그래서 생각한 방법은
1) 같은 원소는 또 확인할 필요가 없으니 집합(set)객체를 만들고
2) 주어진 배열에 대해 반복문을 돌면서
3) 원소에 곱하기2나 나누기2를 한 값이 set에 없다면 add,
4) 있다면 바로 True를, 반복문이 종료될 때 까지 없다면 False를 리턴해주기로 했다.

이렇게 다시 풀어서 시간복잡도를 절반으로 줄일 수 있었다.
끝 !

👨‍💻 소스 코드

class Solution(object):
    def checkIfExist(self, arr):
        """
        :type arr: List[int]
        :rtype: bool
        """
        temp  = set()
        temp.add(arr[0])
        for a in arr[1:]:
            if a*2 in temp or a/2.0 in temp:
                return True
            else:
                temp.add(a)
   
        return False