首页 > 代码库 > Two Sum
Two Sum
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
1 class Solution { 2 public: 3 vector<int> twoSum(vector<int> &numbers, int target) { 4 bool find = false; 5 int index1,index2; 6 auto it = numbers.begin(); 7 auto tmp = it; 8 while(!find && it != numbers.end() ){ 9 tmp = it+1;10 while(!find && tmp!=numbers.end() ){11 if(*it+*tmp ==target){12 find = true;13 index2 = tmp - numbers.begin()+1;14 }15 ++tmp;16 }17 ++it;18 }19 index1 = it-numbers.begin();20 vector<int> temp;21 if(it != numbers.end()){22 temp.push_back(index1);23 temp.push_back(index2);24 }25 26 return temp;27 }28 };
这段代码时间复杂度为O(n^2),,在本地测试通过,但是oj不能ac,错误信息:Time Limit Exceeded超时了。
使用sort (algorithm)函数 先对 numbers的拷贝 排序(O(NlogN)),找出 add number ,存放在 vector<int>中。
在走一遍 vector<int> numbers (O(N)),记下索引,有些小细节注意一下就OK了。
class Solution {public: vector<int> twoSum(vector<int> &numbers, int target) { vector<int> index; vector<int> copy(numbers); sort(copy.begin(),copy.end());//O(nlogn) vector<int> returnnu(SumAlgorithm(copy,target)); int i=0,j=0,flag=-1;//flag 用于标记 index里边是否有数了,防止 两数一样,如果flag初始化0 ,会出错。 while(j<2 && !returnnu.empty() && i<= numbers.size()-1){ if(i != flag && returnnu[j] == numbers[i]){ index.push_back(1+i); flag = i; i=-1;//非常重要,纠结这里好久了,numbers没排序,i需要置0 ++j; } ++i; } if( index[0] > index[1]){ index[1] = index[1]^index[0]; index[0] = index[0]^index[1]; index[1] = index[1]^index[0]; } return index; } //返回 两个 add的数 ,有序 vector<int> SumAlgorithm(vector<int>& numbers,int target){ vector<int> tmp; int j = numbers.size()-1; int i=0,sum =0; bool find = false; while(!find && i<j){ sum = numbers[i] + numbers[j]; if(sum == target) find = true; else if(sum < target) ++i; else --j; } if(find){ tmp.push_back(numbers[i]); tmp.push_back(numbers[j]); } return tmp; }};
Two Sum
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。