【字节跳动二面】在数组中随机选2个数使得和为K,输出所有的组合

题目:给一个数组(无序),在数组中随机选2个数使得和为K,输出所有的组合。

心理活动:面试时看到这个题第一想法就是给数组两个标志位,一头一尾,然后通过判定两个数的和,如果比target大,尾--,如果比target小,头++;这个想法是好的,但是如果中间出现了重复的值就会计算错误,所以要是想处理重复元素的问题一定要对次数下手,所以面试结束后重新对这个题做了一遍。

分析:首先需要将数组排序,然后借助一个辅助的HashMap用来存储元素和出现的次数。每次遍历的时候从小到大进行遍历,如果出现了就将两个数打印出来,否则就存储map然后改变出现次数。

    public void twoSum(int[] nums, int target) {
        //判定边界
        if (nums == null && nums.length == 0) return;
        //定义一个hashmap,key的地方存储元素,value存储出现的次数
        HashMap 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,到这类你也应该能看懂测试用例了把。

因为这个方法也是我自己刚想的,如果不对可以评论区讨论。

你可能感兴趣的:(Java)