暴力法
思路:用嵌套的 for 循环两次遍历数组,找到相加之和为指定值的两个数组元素的索引。
C 代码:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* twoSum(int* nums, int numsSize, int target, int* returnSize) {
*returnSize = 2;
int* res = (int*)malloc(sizeof(int) * (*returnSize)); // 给返回值数组划分好空间
// 可以假设每种输入只会对应一个答案,
// for 循环可以不加 break。
for (int i = 0; i < numsSize; i++) {
// 数组中同一个元素在答案里不能重复出现,
// 因此 j 从 i+1 开始。
for (int j = i + 1; j < numsSize; j++) {
if (nums[i] + nums[j] == target) {
res[0] = i;
res[1] = j;
}
}
}
return res;
}
另一种写法:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* twoSum(int* nums, int numsSize, int target, int* returnSize) {
*returnSize = 2;
int* res = (int*)malloc(sizeof(int) * (*returnSize));
// 可以假设每种输入只会对应一个答案,
// for 循环可以不加 break。
for (int i = 0; i < numsSize; i++) {
// 数组中同一个元素在答案里不能重复出现,
// 因此 j 从 i+1 开始。
for (int j = i + 1; j < numsSize; j++) {
if (nums[i] + nums[j] == target) {
res[0] = i;
res[1] = j;
return res; // 代码一定在这里结束
}
}
}
return NULL; // 代码不可能执行到这一步,加上这一句仅仅是为了代码的完整性。如果不加,程序会报错。
}
改进之后:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
int* twoSum(int* nums, int numsSize, int target, int* returnSize) {
for (int i = 0; i < numsSize; i++) {
for (int j = i + 1; j < numsSize; j++) {
if (nums[i] + nums[j] == target) {
int* res = malloc(sizeof(int) * 2);
res[0] = i;
res[1] = j;
*returnSize = 2;
return res;
}
}
}
*returnSize = 0;
return NULL;
}
Java 代码:
class Solution {
public int[] twoSum(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] + nums[j] == target) {
int[] res = new int[2];
res[0] = i;
res[1] = j;
return res;
}
}
}
return null;
}
}
改进:
class Solution {
public int[] twoSum(int[] nums, int target) {
int n = nums.length;
for (int i = 0; i < n; ++i) {
for (int j = i + 1; j < n; ++j) {
if (nums[i] + nums[j] == target) {
return new int[]{i, j};
}
}
}
return new int[0];
}
}
暴力法的时间复杂度为 O ( n 2 ) O(n^{2}) O(n2)。其中 n 是数组中的元素数量。最坏情况下数组中任意两个数都要被匹配一次。空间复杂度: O ( 1 ) O(1) O(1)。