Two Sum - LeetCode 1

来源:互联网 发布:张家港淘宝拍摄 编辑:程序博客网 时间:2024/06/16 23:50

题目描述:

Given an array of integers, find two numbers such that they add up to a specific target number.

The function twoSum should return indices of the two numbers such that they add up to the target, where index1 must be less than index2. Please 

note that your returned answers (both index1 and index2) are not zero-based.

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

Input: numbers={2, 7, 11, 15}, target=9
Output: index1=1, index2=2

Hide Tags Array Hash Table

分析:

查找满足条件的两个数,是一类很经典的题型,也有多种变形。《编程之美》中有详细讲解,其中一种针对已排序的数组的处理方法是用双指针,两边往中间靠

拢进行搜索,若两者之和大于目标,则右指针左移,若两者之和小于目标,则左指针右移,若相等,则返回两个加数。此题中的输入数组是未排序的,而且要求输出的

处理思路:
方法一、由于要输出下标,因此在排序之前得把下标和元素绑在一起,这可通过定义一个结构体实现,然后按照元素值递增排序,最后再利用双指针,两头向中间扫
描,若符合条件,则输出元素下标。此法时间复杂度为nlog(n)

方法二、此题还可以先将元素和其下标映射到一个multimap(允许重复key),由于map自带按key非递减排序功能,因此可以两头搜索,若得到满足条件的加数,返回对应的下标即可。

以下是C++实现代码:

/*///方法一////16ms/////*/struct entry{    int num;    int index;};bool cmp(entry a, entry b){       return a.num < b.num;}class Solution {public:    vector<int> twoSum(vector<int>& nums, int target) {        vector<int> rev;        int size= nums.size();        if(size < 2)            return rev;                vector<entry> vec;                for(int i = 0; i< size; i++)        {            entry en;            en.num = nums[i];            en.index = i+1;            vec.push_back(en);        }                sort(vec.begin(),vec.end(),cmp);                int i = 0,j = size-1;        while(i < j )        {            int sum = vec[i].num + vec[j].num ;            if(sum == target)            {                if(vec[i].index < vec[j].index)                {                    rev.push_back(vec[i].index);                    rev.push_back(vec[j].index);                }                else                {                    rev.push_back(vec[j].index);                    rev.push_back(vec[i].index);                }                break;            }            else if(sum < target)                i++;            else                 j--;        }        return rev;    }};


/*//方法二/////////////////28ms///////*/

class Solution {public:    vector<int> twoSum(vector<int>& nums, int target) {         vector<int> rev;        int size= nums.size();        if(size < 2)            return rev;                multimap<int,int> m;        for(int i = 0; i< size; i++)        {            m.insert(make_pair(nums[i],i+1));        }                 multimap<int,int>::iterator lef = m.begin();         multimap<int,int>::iterator rig = m.end();         rig--;         while(lef != rig)         {            if(lef->first + rig->first == target)            {                int a = lef->second;int b = rig->second;if(a < b){rev.push_back(a);rev.push_back(b);}else{rev.push_back(b);rev.push_back(a);}                break;            }            else if(lef->first + rig->first < target)                lef++;            else                rig--;        }        return rev;    }};




0 0