题目
- 两数之和 II - 输入有序数组
给定一个已按照升序排列 的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
返回的下标值(index1 和 index2)不是从零开始的。
你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
思路
这道题是easy,思路也的确挺简单的。
因为这个数组已经是升序的了,又因为题目已经保证答案存在且唯一,所以可以直接用双下标法,一个从小到大,一个从大到小。当两个下标对应的数相加之和大于目标值时,把右边的下标减1,反之则左边的加1,直到找到两个下标对应的数相加之和恰好等于目标值为止。
编程
针对算法的代码没有什么特别的技巧。主要是跑起这个函数用的代码有点要求。这道题我用了C语言来解题,函数原型是:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* twoSum(int* numbers, int numbersSize, int target, int* returnSize)
要注意这个返回值是要在自己写的函数里的malloc的,而第三个参数returnSize的含义是给调用者知悉,malloc到底分配了多少个sizeof(int)。如果没有正确传参,就会报类似如下的错误:
Char 3: runtime error: load of null pointer of type ‘int’ (Serializer.c)
代码
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* twoSum(int* numbers, int numbersSize, int target, int* returnSize){
int ix1 = 0;
int ix2 = numbersSize - 1 ;
int * returnSizeTmp = (int *)malloc(2*sizeof(int));
while(1){
int currSum = numbers[ix1] + numbers[ix2];
if (currSum == target){
returnSizeTmp[0] = ix1+1;
returnSizeTmp[1] = ix2+1;
break;
}
else if (currSum > target){
ix2--;
}else{
ix1++;
}
}
*returnSize = 2;
return returnSizeTmp;
}
扩展
如果答案不是唯一的,要输出所有可能的组合。
如果答案可能没有,要输出“mission impossible”。