167. 两数之和 II - 输入有序数组

[AmelloAster]

167. 两数之和 II - 输入有序数组

给定一个已按照升序排列 的有序数组，找到两个数使得它们相加之和等于目标数。

函数应该返回这两个下标值 index1 和 index2，其中 index1 必须小于 index2。

说明:

返回的下标值（index1 和 index2）不是从零开始的。 你可以假设每个输入只对应唯一的答案，而且你不可以重复使用相同的元素。 示例:

输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。

解题代码

var twoSum = function(numbers, target) {
    let i= 0;
    let mid = Math.floor(numbers.length / 2);
    let left = numbers.slice(0, mid);
    let right = numbers.slice(mid, numbers.length);
    let leftEnd = left[left.length - 1];
    let rightEnd = right[0];
    let num = target > rightEnd ? rightEnd : leftEnd;
    let dir = '';

    if (num === leftEnd) {
        dir = 'left'
        i = left.length - 1
    } else {
        dir = 'right'
    }

    while ((dir === 'left' && i >= 0) || (dir === 'right' && i < right.length)) {
        let lIdx = left.findIndex(m => m === target - num);
        let rIdx = right.findIndex(m => m === target - num);

        if (lIdx === i && dir === 'left') {
            lIdx = -1
        }
        if (rIdx === i && dir === 'right') {
            rIdx = -1
        }
        if (dir === 'left') {
            if (lIdx === -1 && rIdx !== -1) {
                return [i + 1, left.length + rIdx + 1]
            }
            if (lIdx !== -1 && rIdx === -1) {
                return [lIdx + 1, i + 1]
            }

            if (lIdx === -1 && rIdx === -1) {
                i--
                num = left[i]
            }
        }

        if (dir === 'right') {
            if (lIdx === -1 && rIdx !== -1) {
                return [i + left.length + 1, i + left.length + rIdx + 1]
            }
            if (lIdx !== -1 && rIdx === -1) {
                return [lIdx + 1, i + left.length + 1]
            }

            if (lIdx === -1 && rIdx === -1) {
                i++
                num = right[i]
            }
        }
    }

}

解题思路

1. 二分
2. 判断 target 是应该从左边开始还是右边开始
3. 确定方向后开始循环 在左边和右边的数组里查找 target-num 的目标元素
4. 找到元素后根据不同条件返回值

代码效率

执行用时：
104 ms, 在所有 JavaScript 提交中击败了25.91%的用户
内存消耗：
38.3 MB, 在所有 JavaScript 提交中击败了10.00%的用户