LeetCode算法问题9 —— Two Sum

先看问题描述

题目给定一个整数型数组以及一个目标和，要求找出数组中和为目标和的整数对(两个整数相互独立)。特别的，题目假定输入限制了数组中有切仅有一对整数答案。

首先想到的办法肯定是对于数组进行遍历，在这一层遍历的基础上再建立一个向后的遍历以查找答案，它的基本思想是这样的：

in array a1, a2, ..., an
for i from 1 to n-1
    for j from i+1 to n
        if (ai + aj == target)
            return <ai, aj>

这样的做法很明显效率较差，两层循环造成了O(n²)的复杂度。

因此，我利用了STL的map，它有find函数可以较快地找到我需要的数据：
1. 将数组遍历，将具体值与其下边组成数据对放入map中
2. 再次遍历数组，对于每一个整数，在map中查找target与该整数的差。如果存在这个数且相互独立(两个数下标不同)，则找到了答案。

具体代码如下

 vector<int> twoSum(vector<int>& nums, int target) {
        map<int, int> findAnswer;
        map<int, int>::iterator findB;
        for (int i = 0; i < nums.size(); i++)
            findAnswer[nums[i]] = i;
        for (int i = 0; i < nums.size(); i++) {
            int temp = target - nums[i];
            findB = findAnswer.find(temp);
            if (findB != findAnswer.end() && findB->second != i) {
                vector<int> answer(2);
                answer[1] = findB->second;
                answer[0] = i;
                return answer;
            }
        }
    }