leetcode第一题两数之和击败了 98.11% 的用户的答案（C++）

虽然题目简单，但我这好不容易优化到前2%，感觉也值得分享给大家（方法比较偷机）

 

题目：

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

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

示例:

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

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

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

 

我的解答：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res(2);
        int endPos = nums.size();
        //vector内存是连续的 这里直接取地址
        //这样后面访问时不需要调用vecotr的成员函数
        //因为不清楚编译器优化级别
        int *numArr = &nums[0];

        if (endPos < 5)
        {
            //数组长度比较小时使用原始的双循环法更快点
            for (int i = 0; i < endPos; ++i)
            {
                //遍历数组，找出每个元素与target之差做为寻找目标
                int nNeed = target - numArr[i];
                for (int j = i + 1; j < endPos; ++j)
                {
                    //在后面找，看有没有与目标相同的数字
                    if (numArr[j] == nNeed)
                    {
                        //如果有直接返回
                        res[0] = i;
                        res[1] = j;
                        return res;
                    }
                }
            }
        }

        //数组比较大时使用一次遍历哈希查找的方法比较快
        map<int, int> mpNums;
        pair<map<int, int>::iterator, bool> pairRet;
        //int numCurr;

        //遍历数组
        for (int i = 0; i < endPos; ++i)
        {
            //把当前数值当key，当前位置当value插入map
            //numCurr = numArr[i]; //实验发现这里使用numCurr取值代替numArr[i]反而慢了
            //看来读取消耗远低于写
            pairRet = mpNums.insert(make_pair(numArr[i], i));

            //如果插入成功（大部分情况下是插入成功的）
            if (pairRet.second)
            {
                //查看map里面是否有目前值-当前元素值的数据存在
                //如果有就说明找到目标
                int numNeed = target - numArr[i];
                map<int, int>::iterator it = mpNums.find(numNeed);
                if (it != mpNums.end() && it->second != i)
                {
                    //题目要求不能重复使用自己，所以需要限制it->second != i
                    res[0] = it->second;
                    res[1] = i;
                    return res;
                }
            }
            else
            {
                //如果插入失败说明
                //已经在map存在相同数值，则看它们加起来是否等于target
                if ((numArr[i] << 1) == target) //2 * numArr[i]
                {
                    res[0] = pairRet.first->second;
                    res[1] = i;
                    return res;
                }
            }
        }

        res.clear();
        return res;
    }

};

 

执行结果：

通过

显示详情

执行用时 :8 ms, 在所有 cpp 提交中击败了98.11%的用户

内存消耗 :10.1 MB, 在所有 cpp 提交中击败了37.46%的用户