本道题题意比较好理解,就是在数组中找当前数的二倍或者一半出现,如果一个都没有,那么就返回false,这个时候我们就要来分析,是找二倍还是找一半,如果当前数存在二倍和一半,是选择二倍还是选择一半?
假设当前的数是4,数组中有2,8,16,我们可以看到,4的二倍,一半都在数组中,如果我们选择了2,刚好2,4凑一对,8 16凑一对,如果我们选择了8,那么只有4,8凑一对,剩下的2,16就落下了,所以我们可以总结出来,需要先从小到大进行排序,如果是负数需要找当前的数的一半,因为负数乘2越乘越小,如果是正数的话,需要找当前数的二倍。
首先我们需要使用TreeMap,key是数值,value是该数字出现的次数,因为需要对key进行排序,所以要使用TreeMap,然后接着我们需要遍历keySet(),需要先判断当前的key是正数还是负数,还要拿到当前数的value,还要拿到目标值。
需要拿到的东西都拿到以后,就开始进行判断,如果目标值不存在的话,直接返回false,如果当前的值对应的value小于目标值对应的value,返回false,如果等于,这个目标值就可以不需要了,直接从map中移出,如果大于,则需要将目标值的value减去当前值的value。我们不仅需要移出目标值,也需要移出当前值,因为最后判断的结果是根据map.size()是否等于0来看的。
class Solution {
public boolean canReorderDoubled(int[] arr) {
//使用TreeMap来将key进行排序
TreeMap<Integer,Integer> map = new TreeMap<>();
for(int num:arr){
map.put(num,map.getOrDefault(num,0)+1);
}
Iterator<Integer> iterator = map.keySet().iterator();
//遍历key
while(iterator.hasNext()){
//need是需要的数,如果key是小于0,那他需要的就是/2,如果key是大于0,那他需要的就是*2
int key = iterator.next(),count = map.get(key),need = key < 0? key/2:key*2;
//因为除可能会向下取整,所以得先检验下
if(need < 0 && need*2 != key) return false; //[-5,-2]应该是不满足的
//如果不存在需要的数,就返回false
if(map.get(need) == null) return false;
int needcount = map.get(need) ;
//如果需要的数量比当前数值的小,则直接返回false
if(needcount < count){
return false;
}
if(needcount == count){
map.remove(need);
}else{
map.put(need,needcount-count);
}
map.remove(key);
iterator = map.keySet().iterator();
}
return map.size() == 0;
}
}