原题链接
题目描述
判断数组中是否存在这样两个数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;
}
}