【LeetCode】（数组）-----0001.两数之和

昨天晚上没来得及做，今天补上

描述：

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

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

 

示例：

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

//自己的代码
class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer, Integer> mp = new HashMap<>();
        for(int i=0; i<nums.length; i++){
            mp.put(i, nums[i]);
        }
        Arrays.sort(nums);
        int p = 0, q = nums.length-1;
        int ant[] = new int[2];
        List<Integer> one = new ArrayList<>();
        List<Integer> two = new ArrayList<>();
        while(p<q){
            if(nums[p] + nums[q] == target) {
                for (int key: mp.keySet()){
                    if(mp.get(key)==nums[p]) one.add(key);
                    if(mp.get(key)==nums[q]) two.add(key);
                }
                break;
            }
            if(nums[p] + nums[q] > target) q--;
            if(nums[p] + nums[q] < target) p++;
        }
        if(one.size() == 1 && two.size() == 1){
            ant[0] = one.get(0);
            ant[1] = two.get(0);
        }
        else {
            ant[0] = Math.min(one.get(0),one.get(1));
            ant[1] = Math.max(one.get(0),one.get(1));
        }
        return ant;
    }
}

//最优代码
class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int complement = target - nums[i];
            if (map.containsKey(complement)) {
                return new int[] { map.get(complement), i };
            }
            map.put(nums[i], i);
        }
        throw new IllegalArgumentException("No two sum solution");
    }
}

用到了map容器，但不清楚map有containsKey的方法，所以绕了很大一个圈子。试图先排序后线性遍历，再匹配索引，太复杂了。然后我就算知道有containsKey也写不出这么好好看的代码，我个人的习惯是先插入后比对，这个习惯不好。

最后这里面有一个小小的坑，测试用例是会出现重复值的，我看到题目描述中说每种输入只对应一个答案就以为不会有重复值，WA了一次。[2,2,3] 4这样的测试用例我可以接受，是自己审题不清，但如果有[2,2,3] 5这样的测试用例那只能说题目有问题了。