LeetCode刷题笔记-1.两数之和

C代码

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
struct cmhash_st {
    int key;
    int value;
    UT_hash_handle hh;
};
int* twoSum(int* nums, int numsSize, int target, int* returnSize){
    struct cmhash_st *hmap = NULL;
    struct cmhash_st *entry = NULL;
    int i;
    int num;
    int *ret = NULL;
    //建立hash表
    for(i = 0; i < numsSize; i++) {
        entry = calloc(1, sizeof(struct cmhash_st));
        entry->key = nums[i];
        entry->value = i;
        HASH_ADD_INT(hmap, key, entry);
        entry = NULL;
    }
    //查询值
    for(i = 0; i < numsSize; i++) {
        num = target - nums[i];
        HASH_FIND_INT(hmap, &num, entry);
        if(entry != NULL && entry->value != i) {
            ret = calloc(2, sizeof(int));
            ret[0] = i;
            ret[1] = entry->value;
            *returnSize = 2;
            return ret;
        }
    }
    return ret;
}

注意点

1. 需要判断不能重复取，即需要判断entry-value != i的场景；
2. 注意设置返回长度*returnSize = 2
3. 内存使用需要优化（待续）

结果

Python3

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        hmap = {}
        for i,val in enumerate(nums, 0):
            hmap[val] = i
        for i, val in enumerate(nums, 0):
            if target - val in hmap:
                if hmap[target - val] != i:
                    return [i, hmap[target - val]]
        return []

学习点

1. 使用enumerate(List[int],start_index)函数来遍历方便获取数组下标；
2. python中的字典的key可以为数字；

python执行结果

题目