https://leetcode-cn.com/problems/two-sum/
给定一个整数数组nums
和一个整数目标值target
,在该数组中找出和为目标值target
的那两个整数,并返回它们的数组下标。只有一个答案
题目说只有唯一确定的解(你可以假设每种输入只会对应一个答案
),又能按区间将数组排序,所以想到了双指针。
有三种思路,分别是:暴力、双指针、哈希表
先说暴力,双重循环,找到答案后结束。这样写的时间复杂度很高。
再说双指针做法:先将数组按非递减的顺序排好序,左右指针向中间靠拢,当两个指针指向的数字之和等于target
时返回结果;大于target
时右指针左移;小于target
时左指针右移。
最后说下哈希表,我们的目的是找到数组中存在的两个目标元素。每次访问到一个数\(x\)时查询\(target - x\)之前有没有出现过,有的话返回下标,没有的话将\(x\)加入哈希表。
暴力
class Solution { public: vector<int> twoSum(vector<int>& nums, int target) { int i, j; //双重循环 for(i = 0; i < nums.size(); ++i){ for(int j = i + 1; j < nums.size(); ++j){ if(nums[i] + nums[j] == target) return {i, j}; } } return {-1, -1}; } };
双指针 用双指针的前提是有序,一定不能忘记排序
class Solution { public: vector<int> twoSum(vector<int>& num, int target) { //新开一个数组复制原数组 新数组排序后打乱下标,原数组提供下标 vector<int>nums = num; //排序 sort(nums.begin(), nums.end()); int n = nums.size(); //左右双指针 int i = 0, j = n - 1; int sum ; while(i < j) { sum = nums[i] + nums[j]; //满足条件 if(sum == target) { i = nums[i]; j = nums[j]; break; } //之和小于目标值,让左指针右移 if(sum < target) ++ i; //之和大于目标值,让右指针左移 else -- j; } if(i > j) swap(i, j); vector<int> ans(2, -1); //用的很笨的方法在原数组里面查询下标 for(int k = 0; k < n; ++ k) { if(num[k] == i) { ans[0] = k; break; } } for(int k = 0; k < n; ++ k) { if(k == ans[0]) continue; if(num[k] == j) { ans[1] = k; break; } } return ans; } };
哈希表
class Solution { public: vector<int> twoSum(vector<int>& nums, int target) { unordered_map<int, int>m; for(int i = 0; i < nums.size(); ++ i) { auto it = m.find(target - nums[i]); //在哈希表里找到了目标值 if(it != m.end()) { return {i, it->second}; } m[nums[i]] = i; } return {}; } };
这是很多人接触力扣的第一题。愿大家满怀热爱、一往无前。
附一张我的提交记录