2Sum问题是3Sum和4Sum的基础,很多OJ都是以此为最简单的练手题的。
题目描述:
从一个数组里找出两个和为target的数。
LeetCode上的描述:
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
Example:
Given nums = [2, 7, 11, 15], target = 9, Because nums[0] + nums[1] = 2 + 7 = 9, return [0, 1].
主要有三种思路:
1 暴力求解,就是两层循环,每次固定一个数nums[i],找其余的数是否为target - nums[i],显然时间复杂度为O(n2);
2 排序后用两个指针left和right分别从开头和末尾往中间检查,
1)如果nums[left] + nums[right] == target,说明找到两个数,直接返回;
2)如果nums[left] + nums[right] < target, 则left++;
3)如果nums[left] + nums[right] > target, 则right--;
如果left >= right 说明不错在这样的两个数。
该算法的复杂度以排序的为主是O(nlogn);
3 空间换时间 建立一个Map,依次把所有的元素都存进去,存的过程中如果发现map里已经存在target - nums[i], 则直接返回,时间复杂度为O(n),空间复杂度为O(n).
这里给出第三种算法的Java代码:
1 class Solution { 2 public int[] twoSum(int[] nums, int target) { 3 Map map = new HashMap(); 4 for(int i = 0; i < nums.length; i++){ 5 map.put(nums[i], i); 6 } 7 for(int i = 0; i < nums.length; i++){ 8 if(map.containsKey(target - nums[i]) && 9 (int)map.get(target - nums[i]) != i){ 10 return (new int[] { i, (int)map.get(target - nums[i]) }); 11 } 12 } 13 return new int[]{-1, -1}; 14 } 15 }