【Leetcode Easy】1346. Check If N and Its Double Exist

原题链接

题目描述

判断数组中是否存在这样两个数N和M，其中N = 2 * M

我的思路

看起来很简单的一道题，但卡了挺久的……果然一段时间不学习就会生疏
一开始的想法是准备一个Set，首先将array排序，（从大到小）遍历排序后的array，判断set中是否存在当前元素的两倍的数据，如果有，返回true，如果没有，将当前数据加入array。
这个思路的问题在于他要求必须是N先加入set，在正数部分是可以的，但遍历过程中，如存在(-10, -20)满足条件，-10先加入set，等到遍历到-20时就会判断错误

改进：

多一个判断条件，遍历到M时，不仅判断set中是否存在2*M，还要判断是否存在M/2（注意这里要保证是整除的）

class Solution {
    public boolean checkIfExist(int[] arr) {
        Arrays.sort(arr);
        Set<Integer> set = new HashSet<>();
        
        for (int i = arr.length - 1; i >= 0; i--) {
            int tmp = arr[i];
            if (set.contains(tmp * 2) || ((tmp & 1) == 0 && set.contains(tmp / 2)))
                return true;
            set.add(tmp);
        }
        return false;
    }
}