一、问题描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9,所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
二、解题思路
首先,返回数组的下标要么是2个要么是0个,当目标值为0或者数组元素个数为0时返回值为0个,其他情况返回值均为2个。因此,在对返回值动态分配空间时直接分配2个int型的空间即可。
在数组中选取两个数的和为目标值时,可以假定第一个数为返回的第一个下标即a[0]=0,然后依次从第二个数开始判断是否存在一个数与nums[a[0]]的和与目标值target相等,如果存在则另a[1]等于该数的下标值,如果不存在让a[0]++,以此类推知道找到符合要求的两个下标并返回。
三、C语言程序
int* twoSum(int* nums, int numsSize, int target, int* returnSize)
{
//分配空间
int* a = (int*)malloc(sizeof(int)* 2);
int i, j;
for (i = 0; i <= numsSize - 2;i++)
for (j = i+1; j <= numsSize - 1; j++)
{
if (nums[i] + nums[j] == target)
{
a[0] = i;
a[1] = j;
*returnSize = 2;
return a;
}
}
//如果上边没有返回,则表明没有找到结果,此时returnSize = 0,返回值也为0
*returnSize = 0;
return 0;
}
四、总结
本题采用了暴力法进行解决,此方法时间复杂度(O(n^2))比较高但是容易理解。需要注意的是分配空间时不要分配太多的空间,最多需要两个int型的空间足以存储返回值,同时还需要注意如果找到满足要求的下标时,要讲returnSize = 2,没找到要使returnSize = 0且返回0.