两数求和问题

问题

/**
 * Given an array of integers, find two numbers such that they add up to a specific target
 * number.
 * The function twoSum should return indices of the two numbers such that they add
 * up to the target, where index1 must be less than index2. Please note that your returned
 * answers (both index1 and index2) are not zero-based.
 * For example:
 * Input: numbers={2, 7, 11, 15}, target=9
 * Output: index1=1, index2=2
 * <p>
 * 给定一个整数数组，找出两个加起来等于特定目标数的数字。
 * 函数twoSum应该返回这两个数字的下标，使它们与目标数相加，其中index1必须小于index2。
 * 请注意，您返回的答案(包括index1和index2)不是从零开始的。
 * <p>
 * 
 */

解法

朴素解法

这个问题很简单。我们可以简单地检查这个整数数组中每个可能的数字对。

最坏情况下的时间复杂度:O(nˆ2)。

public static int[] twoSum1(int[] numbers, int target) {
        int[] ret = new int[2];
        for (int i = 0; i < numbers.length; i++) {
            for (int j = i + 1; j < numbers.length; j++) {
                if (numbers[i] + numbers[j] == target) {
                    ret[0] = i + 1;
                    ret[1] = j + 1;
                }
            }
        }
        return ret;
    }

更好的解法

我们使用HashMap来存储需要的值，value是它的index；

public static int[] twoSum2(int[] numbers, int target) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int[] result = new int[2];

        for (int i = 0; i < numbers.length; i++) {
            if (map.containsKey(numbers[i])) {
                int index = map.get(numbers[i]);
                result[0] = index + 1;
                result[1] = i + 1;
                break;
            } else {
                // key为第i个位置需要的值
                map.put(target - numbers[i], i);
            }
        }
        return result;
    }

时间复杂度取决于HashMap的put和get操作，这是正常的O(1)。

这种方法的时间复杂度为O(n)。