题目:给一个数组(无序),在数组中随机选2个数使得和为K,输出所有的组合。
心理活动:面试时看到这个题第一想法就是给数组两个标志位,一头一尾,然后通过判定两个数的和,如果比target大,尾--,如果比target小,头++;这个想法是好的,但是如果中间出现了重复的值就会计算错误,所以要是想处理重复元素的问题一定要对次数下手,所以面试结束后重新对这个题做了一遍。
分析:首先需要将数组排序,然后借助一个辅助的HashMap用来存储元素和出现的次数。每次遍历的时候从小到大进行遍历,如果出现了就将两个数打印出来,否则就存储map然后改变出现次数。
public void twoSum(int[] nums, int target) {
//判定边界
if (nums == null && nums.length == 0) return;
//给数组从小到大排个序
Arrays.sort(nums);
//定义一个hashmap,key的地方存储元素,value存储出现的次数
HashMap<Integer, Integer> hashMap = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
//如果找到符合条件的值就打印相应的次数
if (hashMap.containsKey(target - nums[i])) {
for (int j = 0; j < hashMap.get(target - nums[i]); j++) {
System.out.println(target - nums[i] + ":" + nums[i]);
}
}
//如果发现重复元素,次数增加1;如果没有则加入hashmap
if (hashMap.containsKey(nums[i]))
hashMap.put(nums[i], hashMap.get(nums[i]) + 1);
else
hashMap.put(nums[i], 1);
}
}
/**
* 输入:{1, 1, 1, 1}, 2
* 输出:1:1
* 1:1
* 1:1
* 1:1
* 1:1
* 1:1
* 输入:{1, 2, 2, 7, 7, 9, 8} ,9
* 输出:2:7
* 2:7
* 2:7
* 2:7
* 1:8
* */
如果你有疑问在这两个测试用例,为什么会出现重复的值你可以这么想:比如我有一个数组{1,1,1},我另target是2,那么我随机选取两个数的选择方式有几种呢?一定是三种,且都是(1,1),一组是前两个1,一组是后两个1,最后一组是第一个和第三个1,到这类你也应该能看懂测试用例了把。
因为这个方法也是我自己刚想的,如果不对可以评论区讨论。