我就在LeetCode自带的编辑器写的代码,还挺好用的,这题的思路我一开始碰着试一试的态度用一个数组标记,初始化为0,存在就为1,后来发现这样不行,因为题目也没给范围,后来交了一发运行错误,大概是因为数组越界了,所以就用了map标记,key就是个数,然后遍历数组,看target - nums[i]是否存在,一开始没有理解题目说的“不能重复利用这个数组中同样的元素” , 后来交了一发,看到了错误案例才懂,大概就是要考虑到 target - nums[i] == nums[i] 但是nums[i]的个数又只有一个的情况,最后修改了一下代码才过,这里贴一下代码。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int i , j;
// int flag[1000000];
// fill( flag , flag + 100000 , 0 );
int map_flag = 0;
map<int , int>flag;//标记是否出现这个数
for( i = 0 ; i < nums.size() ; i++ ){
flag[nums[i]]++;//个数即为key
}
vector<int>ans;
int mark = 0;
for( i = 0 ; i < nums.size() ; i++ ){
if( flag[target - nums[i]] ){//存在和nums[i]相加为target的数
if( target - nums[i] == nums[i] ){//这个数和本身相等
if( flag[nums[i]] != 1 ){//有多个
ans.push_back(i);
for( j = i + 1 ; j < nums.size() ; j++ ){
if( nums[j] == nums[i] ){
ans.push_back(j);
break;
}
}
}
else continue;//这个数只有一个
}
else{//这个数和本身不相等
ans.push_back(i);
mark = target - nums[i];
for( i = 0 ; i < nums.size() ; i++ ){
if( mark == nums[i] ){
ans.push_back(i);
break;
}
}
}
break;
}
}
return ans;
}
};
我写这篇博客的时候还没有看官方题解,我的时间击败了百分之五十多的c++过题代码,但是我没有优化,应该是可以把第二重for循环提出来的,这就是我第一题的思路,有错误或者不足的地方希望大家指正。
补:看了看自己的代码 ,好像第二重循环提不提出来都是一样,都是扫一遍,看了官方题解,然后发现中间写的有点冗余 , 应该可以直接用 下标 作为key , 这样的话中间就不用if讨论那么多了。
后来的ac代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int i , j;
// int flag[1000000];
// fill( flag , flag + 100000 , 0 );
int map_flag = 0;
map<int , int>flag;
for( i = 0 ; i < nums.size() ; i++ ){
flag[nums[i]] = i;
}
vector<int>ans;
int mark = 0;
for( i = 0 ; i < nums.size() ; i++ ){
if( flag[target - nums[i]] && ( flag[target - nums[i]] != i ) ){
ans.push_back(i);
mark = target - nums[i];
ans.push_back(flag[mark]);
break;
}
}
return ans;
}
};