【LeetCode1】两数之和(哈希表)

一、题目

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

示例 1：

输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
示例 2：

输入：nums = [3,2,4], target = 6
输出：[1,2]
示例 3：

输入：nums = [3,3], target = 6
输出：[0,1]

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/two-sum

二、基于哈希表的算法 

算法描述

• 在遍历数组的时候，就把键值对信息放入哈希表中，并在这个哈希表中查找target-num[i],一旦查找出来，说明要找的两个数就存在于这个数组中。

1、定义哈希表

struct hashTable {
    int key;
    int val;
    UT_hash_handle hh;
};

UT_hash_handle表示结构体类型为哈希表。

2、查找哈希表中是否存在某值

struct hashTable* find(int ikey) {
    struct hashTable* tmp;
    HASH_FIND_INT(hashtable, &ikey, tmp);
    return tmp;
}

HASH_FIND_INT，第一个参数哈希表指针，第二个参数是键值的地址，第三个参数是输出的哈希表指针。当可以在哈希表中找到相应键值时，其返回给定键的结构，当找不到时返回NULL。

3、向哈希表中插入键值对

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;
    }
}

如果要查找的键值不存在的话，就向哈希表中插入该键值对，否则更新键值的下标。 

 4、用哈希表匹配数组的两个数

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;
}

比如，两数之和为9，在数组

1 3 5 6 9 2

中

• 9-1=8，向哈希表中查找8，NULL，把1对应的键值对放入哈希表中
• 9-3=6，向哈希表中查找6，NULL，把3对应的键值对放入哈希表中 
• 9-5=4，向哈希表中查找4，NULL，把5对应的键值对放入哈希表中
• 9-6=3，向哈希表中查找3，哈希表中存在3，返回下标信息

5、完整代码 

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;
}

作者：LeetCode-Solution
链接：https://leetcode.cn/problems/two-sum/solution/liang-shu-zhi-he-by-leetcode-solution/
来源：力扣（LeetCode）

6、复杂度

• 时间复杂度O(n)。相比于暴力算法的两个for，利用哈希表直接将O(n)降为O(1)
• 空间复杂度O(n)。主要为哈希表的开销。