LeetCode 1. Two Sum Java

Posted 2021-11-30 blogxjc

1. Two Sum（两数之和）

 

题目：

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

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

　　示例:　　　

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

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

思路：　　

　　简单的遍历即可，考虑到复杂度可以借助hashmap，只需要图的基本操作即可。

 

代码：

　　代码一（时间消耗量较大）

 1 public static int[] twoSum(int[] nums, int target) {
 2         Map<Integer, Integer> map = new HashMap<>();
 3         for (int i = 0; i < nums.length; i++) 
 4             map.put(nums[i], i);
 5         for (int i = 0; i < nums.length; i++) 
 6         {
 7             int number = target - nums[i];
 8             if (map.containsKey(number)&&map.get(number)!=i) {
 9                 return new int[] { map.get(number), i };
10             }
11         }
12         throw new IllegalArgumentException("No two sum solution");
13     }

View Code

 

　　代码二（优化之后）

 1     public static int[] twoSum(int[] nums, int target) {
 2         Map<Integer, Integer> map = new HashMap<>();
 3         for (int i = 0; i < nums.length; i++) 
 4         {
 5             int number = target - nums[i];
 6             if (map.containsKey(number)) {
 7                 return new int[] { map.get(number), i };
 8             }
 9             map.put(nums[i], i);
10         }
11         throw new IllegalArgumentException("No two sum solution");
12     }

View Code