Leetcode_0001_两数之和

0001 两数之和

题目链接：https://leetcode-cn.com/problems/two-sum/

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9

所以返回 [0, 1]

解法一：暴力破解 时间复杂度O(n^2)

1. c语言版

int* twoSum(int* nums, int numsSize, int target) {
    int * result = (int *) malloc(sizeof(int) * 2);
    for (int i = 0; i < numsSize - 1; i++)
        for (int j = i + 1; j < numsSize; j++)
        {
            if (nums[i] + nums[j] == target)
            {
                result[0] = i;
                result[1] = j;
				break;
            }
        }
    return result;
}

2.java版

public int[] twoSum(int[] nums, int target) {
		int numsSize = nums.length;
	    for (int i = 0; i < numsSize - 1; i++)
	        for (int j = i + 1; j < numsSize; j++)
	        {
	            if (nums[i] + nums[j] == target)
	            {
	                return new int[] { i, j};
	            }
	        }
	    return null;
    }

解法二: 哈希表 时间复杂度O(n)

1. c语言版

由于C不像Java、C++本身包含哈希表等数据结构，所以需要手动实现。但是在LeetCode中，C已经包含一个开源库uthash.h,可以实现哈希表各种操作。uthash的用法可以见我的一篇博客。

typedef struct my_struct {
    int id;            /* we'll use this field as the key */
    int index;
    UT_hash_handle hh; /* makes this structure hashable */
}HashNode;

typedef HashNode* HashHead;
HashNode *find(HashHead head, int key) {
    HashNode *s;
    HASH_FIND_INT(head, &key, s);  /* s: output pointer */
    return s;
}
void add(HashHead *head, HashNode *node) {
    if(!find(*head, node->id))
        HASH_ADD_INT(*head, id, node);
}

int* twoSum(int* nums, int numsSize, int target) {
    int *ans = malloc(sizeof(int) * 2);
    int diff;
    HashHead head = NULL;
    HashNode *node, *addNode;
    // 遍历数组
    for (int i = 0; i < numsSize; i++) 
    {
    	// 找到与target的差值
        diff = target - nums[i];
        // 判断在哈希表中是否有
        node = find(head, diff);
        // 存在的话，因为只有唯一解，所以直接输出结果
        if (node)
        {
            ans[0] = node->index; 
            ans[1] = i;
            return ans;
        }
        // 不存在，添加进哈希表
        addNode = malloc(sizeof(HashNode));
        addNode->id = nums[i];
        addNode->index = i;
        add(&head, addNode);
    }
    return ans;
}

2.java版

public int[] twoSum(int[] nums, int target) {
        HashMap map = new HashMap();
        for (int i = 0; i < nums.length; i++)
        {
            int diff = target-nums[i];
            if (map.containsKey(diff)) {
            	return new int[]{(int) map.get(diff),i};           	
            }
            map.put(nums[i], i);  
        }
        return null;
    }