Leetcode对撞指针
先从一个简单的例子开始,给定一个排序好的数组,int nums[] = {-2, 0, 1, 3},求出两两之和为target(target = 1)的组合。
最开始学习对撞指针非常疑惑的地方就在于,除了-2 + 3 = 1,还有0 + 1 = 1。因为对撞指针解决的往往并不是唯一解问题。
假设l = 0, r = n - 1 = 3,nums[l] + nums[r] == 1,这时唯一正确的处理方法就是l++,r--。
除了组合直接等于target的情况,两个对撞指针都往中间靠近。还有可能遇到的情况是,假设int nums[] = {-2, -2, 0, 1, 3},
即当对撞指针各走一步以后,此时l = 1, r = 2,由于此时nums[l] + nums[r] != target,而且nums[l]元素不变、nums[r]变小,所以此时只能做出++r的决策。
咱们来具体看几个题吧,第一个题是Two Sum,我说的犯的几个错误吧:
1.对nums进行排序了,导致元素忘记了之前的index。所以需要一个nums_bak进行备份。
2.对nums两个元素重复的处理,最简单的可以用两次循环,第一次找到位置,第二次再沿着之前的位置再循环一次。
还有一种思路就是用记忆法。见下面的代码。
class Solution { public: vector<int> twoSum(vector<int>& nums, int target) { vector<int> nums_bak(nums); sort(nums_bak.begin(), nums_bak.end()); int a = 0x7fffffff, b = 0x7fffffff; int n = nums.size(); vector<int> ret; int l = 0; int r = n - 1; int indexA = -1; int indexB = -1; while (l < r) { int sum = nums_bak[l] + nums_bak[r]; if (sum == target) { a = nums_bak[l++]; b = nums_bak[r--]; } else if (sum > target) { --r; } else { ++l; } } if (a == 0x7fffffff) { return ret; } else { for (int i=0; i<nums.size(); ++i) { if (nums[i] == a && i != indexB) { indexA = i; ret.push_back(i); } if (nums[i] == b && i != indexA) { indexB = i; ret.push_back(i); } } } return ret; } };