题目
题解
一、暴力解法
刚刷第一道,暂时只会暴力方法......
对数组从第一个元素开始,对每一个元素从它后面的元素中遍历找满足条件的元素,找到就可以返回了。
int* twoSum(int* nums, int numsSize, int target, int* returnSize){
int i=0,j=0,k=0;
for(i=0;i<numsSize;i++)
{
for(j=i+1;j<numsSize;j++)
{
if(nums[i]+nums[j]==target)
{
int* ret = malloc(sizeof(int) * 2);
ret[0]=i;
ret[1]=j;
*returnSize=2;
return ret;
}
}
}
*returnSize =0;
return NULL;
}
时间复杂度:O(N^2) 空间复杂度:O(1)
二、哈希表
用哈希表,里面存放的键值对为数组元素和其下标。key为nums[i],value为i。
遍历数组 nums,i 为当前下标,每个值都判断map中是否存在 target-nums[i] 的 key 值。如果存在则找到了两个值,如果不存在则将当前的 (nums[i],i) 存入 hash表 中,继续遍历直到找到为止。
这里只需遍历数组一遍,因为两个元素x,y必定是一前一后出现的,先去哈希表中找x,没找到就把x插入表,后面遍历到y就可以在hash表中找到对应的x了。官网题解代码如下:
struct hashTable {
int key;
int val;
UT_hash_handle hh;
};
struct hashTable* hashtable;
struct hashTable* find(int ikey) {
struct hashTable* tmp;
HASH_FIND_INT(hashtable, &ikey, tmp);
return tmp;
}
void insert(int ikey, int ival) {
struct hashTable* it = find(ikey);
if (it == NULL) {
struct hashTable* tmp = malloc(sizeof(struct hashTable));
tmp->key = ikey, tmp->val = ival;
HASH_ADD_INT(hashtable, key, tmp);
} else {
it->val = ival;
}
}
int* twoSum(int* nums, int numsSize, int target, int* returnSize) {
hashtable = NULL;
for (int i = 0; i < numsSize; i++) {
struct hashTable* it = find(target - nums[i]);
if (it != NULL) {
int* ret = malloc(sizeof(int) * 2);
ret[0] = it->val, ret[1] = i;
*returnSize = 2;
return ret;
}
insert(nums[i], i);
}
*returnSize = 0;
return NULL;
}