LeetCode刷题系列--1. 两数之和

题目:

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
 

 思路:两次遍历:
*      1、用一个 Map<Integer,List<Integer> 存放数组中每个元素的的值与下标,即 (num,[index1,index2]),因为数组中可能有多个值相同的元素
*      2、遍历数组,对指定元素num,看是否存在 target - num 的元素,并获取其下标值
*         1)如果 键为 target - num 对应的 链表中存放的下标数大于等于2 ,则说数组中至少有2个元素的值均为 target/2 ,返回下标即可
         比如[3,3,4,5,4]
*      *    6
*         2)如果  键为 target - num 对应的 链表中存放的下标数为1 ,则要判断存放在链表中的下标值是否等于当前元素num 的下标,为了防止 当num == taeget/2 且元素之出现一次的情况
*           比如[3,4,5,1]
*               6
    public int[] twoSum(int[] nums, int target) {
        Map<Integer,List<Integer>> map = new HashMap<>();
        //第一次遍历,记录下数组中元素与其下标
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(nums[i])){
                map.get(nums[i]).add(i);
            }else{
                List<Integer> list = new ArrayList<>();
                list.add(i);
                map.put(nums[i],list);
            }

        }
        int i=0;
        int j=0;
        int[] result = new int[2];
        //第二次遍历,找到与当前元素 num 之和为target的另外一个元素的下标
        for(;i<nums.length;i++){
            if(map.containsKey(target-nums[i])){
                List<Integer>  list = map.get(target-nums[i]);
                if(list.size()>=2){       //如果数组中有两个及以上的元素 num == target/2 ,则直接获取其下标即可
                    result = new int[]{list.get(0), list.get(1)};
                    break;
                }else {
                    j =  list.get(0);
                    if(i==j){         //为了应对 [3,4,5,1]  6 情况
                        continue;
                    }
                    result = new int[]{i, j};
                    break;
                }
            }
        }

        return result;
    }

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章