【LeetCode-1】 Two Sum(C++)

来源：互联网发布：阿里巴巴订单打印软件编辑：程序博客网时间：2024/06/05 18:03

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution.

Example:

Given nums = [2, 7, 11, 15], target = 9,Because nums[0] + nums[1] = 2 + 7 = 9,return [0, 1].

UPDATE (2016/2/13):
The return format had been changed to zero-based indices. Please read the above updated description carefully.

1.直接暴力解决，用两个for循环依次查找。时间复杂度O(n^2)，空间复杂度O(1)

class Solution {public:    vector<int> twoSum(vector<int>& nums, int target) {        int l=nums.size();        vector<int> a(2);        for(int i=0;i<l;i++){            for(int j=i+1;j<l;j++)               if(nums[j]==target-nums[i])                  {                      a[0]=i;                      a[1]=j;                      return a;                  }        }        return a;    }};

2. 用一个无序multimap，之所以用无序是因为无序比有序的快，不用排序。multimap一个键值可以对应多个值。将数组中的整型数和其对应的索引值存储到multimap中，整形数作为键值。然后用一个for循环在multimap中进行查找。时间复杂度O(n)，空间复杂度O(n)。

class Solution {public:    vector<int> twoSum(vector<int>& nums, int target) {        unordered_multimap<int,int> umap;        vector<int> a(2);        for(int i=0;i<nums.size();i++){            umap.insert(pair<int,int>(nums[i],i));        }        for(int i=0;i<nums.size();i++){            int c=target-nums[i];            if(umap.find(c)!=umap.end()&&umap.find(c)->second!=i){//注意查找到的索引值不能与当前索引值相同。                a[0]=i;                a[1]=umap.find(c)->second;                return a;            }        }        return a;            }};

3.边插入边查找。用一个for循环对数组中的每一个数进行查找，对于当前数，现在multimap中检查有没有满足条件的值，如果有，直接返回。如果没有则插入，这样可以不用判断查找到的数的索引是不是当前数的索引，因为当前数还没有插入到multimap中，所以如果能查找，查找到的数的索引一定与当前数的索引不同。

class Solution {public:    vector<int> twoSum(vector<int>& nums, int target) {        unordered_multimap<int,int> umap;        vector<int> a(2);        for(int i=0;i<nums.size();i++){            int c=target-nums[i];            if(umap.find(c)!=umap.end()){                a[0]=umap.find(c)->second;                a[1]=i;                return a;            }            umap.insert(pair<int,int>(nums[i],i));        }        return a;    }};

0 0