两数之和 - Java实现

23. 两数之和

题目描述

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例

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

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

LeetCode地址

https://leetcode-cn.com/problems/two-sum/

代码

import java.util.Arrays;
import java.util.HashMap;
import java.util.LinkedList;
/**
 * Class day23_两数之和 ...
 *
 * @author LiJun
 * Created on 2019/4/8
 */
public class day23_两数之和 {
    static class Solution {
        /**
         * 跑了一百多毫秒，因为linkedlist contains 要遍历
         */
        public int[] twoSum(int[] nums, int target) {
            LinkedList<Integer> linkedList = new LinkedList();
            for (int i = 0; i < nums.length; i++) {
                if (linkedList.contains(nums[i])) {
                    return new int[]{linkedList.indexOf(nums[i]), i};
                }
                linkedList.add(target - nums[i]);
            }
            return null;
        }

        /**
         * 这个就只需要 6ms
         */
        public int[] twoSum1(int[] nums, int target) {
            HashMap<Integer, Integer> map = new HashMap<>();
            for (int i = 0; i < nums.length; i++) {
                if(map.containsKey(target - nums[i])){
                    return new int[]{map.get(target-nums[i]), i};
                }
                if (!map.containsKey(nums[i])) {
                    map.put(nums[i], i);
                }
            }
            return null;
        }
    }

    public static void main(String[] args) {
        int[] arr = {2, 7, 11, 15};
        Solution solution = new Solution();
        System.out.println(Arrays.toString(solution.twoSum1(arr, 9)));
    }

}