LeetCode算法问题9 —— Two Sum

来源:互联网 发布:浙江网络作家协会主席 编辑:程序博客网 时间:2024/06/16 07:43

先看问题描述
这里写图片描述
题目给定一个整数型数组以及一个目标和,要求找出数组中和为目标和的整数对(两个整数相互独立)。特别的,题目假定输入限制了数组中有切仅有一对整数答案。

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

in array a1, a2, ..., anfor 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;            }        }    }
原创粉丝点击