LeetCode1.两数之和

题目

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路

1.土土的原生思路

由于给定的nums是vector类型，所以先用一个copy数组记录原来的数组顺序，然后将原来的数组排序，从头尾向中间遍历。令i=0，j=nums.size()-1：
$$\{\begin{array}{r}nums[i]+nums[j]=target\dots \dots i,j存入ans\\ nums[i]+nums[j]<target\dots \dots \dots \dots i++\\ nums[i]+nums[j]>target\dots \dots \dots \dots j--\end{array}$$
最后在根据ans中存放的数去找出在原来的数组里的下标，这里要注意考虑两个数一样的情况。

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> copy(nums);
        sort(nums.begin(),nums.end());
        vector<int> ans(2);
        int i=0,j=nums.size()-1;
        while(i<j){
            if(nums[i]+nums[j]==target){
                ans[0]=nums[i];
                ans[1]=nums[j];
                break;
            }else if(nums[i]+nums[j]<target)
                ++i;
            else
                --j;
        }
        for(int i=0;i<copy.size();++i){
            if(copy[i]==ans[0]){
                ans[0]=i;
                break;
            }
        }
        for(int i=0;i<copy.size();++i){
            if(copy[i]==ans[1]&&i!=ans[0]){//为了处理两个数相等的情况
                ans[1]=i;
                break;
            }
        }
        return ans;
    }
};

运行结果

2.评论区的思路

使用map索引，将（nums[i],i）插入map中，再遍历查看target-nums[i]是否为map中的键。注意：当nums中存在两个一样的数时，第二个数插入map会覆盖第一个，所以在之后查找target-nums[i]时要注意判断：mapn.find(target-nums[i])!=mapn.end()&&mapn[target-nums[i]]!=i。

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        map<int,int> mapn;
        vector<int> ans;
        for(int i=0;i<nums.size();++i){
            mapn[nums[i]]=i;
        }
        for(int i=0;i<nums.size();++i){
            if(mapn.find(target-nums[i])!=mapn.end()&&mapn[target-nums[i]]!=i){
                ans.push_back(i);
                ans.push_back(mapn[target-nums[i]]);
                return ans;
            }
        }
        return ans;
    }
};

运行结果