力扣第一题 两数之和扩展版

题目

  在《力扣第一题 两数之和》中,我们给出了三种求解力扣第一题两数之和的方法,这里给出一个它的升级版:找出所有两数之和为target的数组下标。

  力扣第一题 两数之和扩展版:给定一个元素个数为N的整数数组 nums 和一个整数目标值 target,请在该数组中找出所有和为 target  的那两个整数,并返回它们的数组下标。假设可以按任意顺序返回答案。但是,数组中同一个元素在答案里不能重复出现。

基于哈希表找出所有两数之和

  首先定义一个对象 Map<String, Integer> map,key为一个由英文逗号分隔的、任意两个数的下标组成的字符串,value为两数之和。然后,遍历数组,将任意两数之和存入map。最后,遍历map,如果value等于target,则返回其key,这个key就是我们想要的数组下标。答案基于《力扣第一题 两数之和》的第三种解法,稍作重构即可,代码如下:

    /**
     * 二次遍历map,为求解所有两数之和等于target的整数做铺垫
     * @param nums
     * @param target
     * @return
     */
    public static List<String> allSolution(int[] nums, int target) {
        Map<String, Integer> map = new HashMap<>();
        for (int i = 0, len = nums.length; i < len; i++) {
            int numI = nums[i];
            for (int j = i + 1; j < len; j++) {
                map.put(i + "," + j, numI + nums[j]);
            }
        }
        // 保存结果集
        List<String> rets = new ArrayList<>();
        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            if (entry.getValue() == target) {
                rets.add(entry.getKey());
            }
        }

        return rets;
    }
    
复杂度分析
  • 时间复杂度:O(N^2))。最坏情况下数组中任意两个数都要被匹配一次。

  • 空间复杂度:O(N)。

posted @ 2021-06-05 20:07  楼兰胡杨  阅读(97)  评论(0编辑  收藏  举报