LintCode 两数之和

来源:互联网 发布:java相关文献 编辑:程序博客网 时间:2024/05/18 07:39

给一个整数数组,找到两个数使得他们的和等于一个给定的数 target。

你需要实现的函数twoSum需要返回这两个数的下标, 并且第一个下标小于第二个下标。注意这里下标的范围是 1 到 n,不是以 0 开头。

Notice

你可以假设只有一组答案。

Example
给出 numbers = [2, 7, 11, 15], target = 9, 返回 [1, 2].
Either of the following solutions are acceptable:
O(n) Space, O(nlogn) Time
O(n) Space, O(n) Time

感谢山丘儿博客地址http://blog.csdn.net/s634772208的文章。
第一种,就两层循环,复杂度O(n2)。

第二种,将数组排序,然后两个指针从两侧分别向中间移动,左侧i右侧j。当和大于target时,j向左移动,当和小于target时,i向右移动,知道相遇或满足要求。复杂度O(nlgn)。按照这个想法写了一下,由于数组中会有重复的元素,而且排序之后下标会乱,所以这种方法没ac,转而使用第三种方法。

第三种,利用哈希表,将数组元素存到哈希表中,每次检查target-numbers[i]的结果是否在哈希表中。复杂度O(n)。关于如何处理可能出现的重复元素:题目假设只有一组答案,那么说明相同的的元素最多出现两次(而且结果是由这两个相同元素构成),那么在正向扫描数组numbers构造玩哈希表之后,可以反向扫描数组,检查target-numbers[i]。

代码(第三种):

public class Solution {    /*     * @param numbers: An array of Integer     * @param target: target = numbers[index1] + numbers[index2]     * @return: [index1 + 1, index2 + 1] (index1 < index2)     */    public int[] twoSum(int[] numbers, int target) {        // write your code here        int[] res=new int[2];        Map<Integer,Integer> m=new HashMap();        for (int i=0;i<numbers.length;i++){            if(!m.containsKey(numbers[i])){                m.put(numbers[i],i);            }        }        for(int i=numbers.length-1;i>=0;i--){            if(m.containsKey(target-numbers[i])){                System.out.println(i);                System.out.println(m.get(target-numbers[i]));                res[0]=m.get(target-numbers[i])+1;                res[1]=i+1;                break;            }        }        return res;    }}

LintCode上的另外几道题,三数之和,最接近三数之和,四数之和(这几道题均不要求下标),都可以用类似的方法求解(第二种,或第三种)。详细参考eudiwffe的文章

原创粉丝点击