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;    
    }
};

 

posted @ 2018-08-29 20:54  ly-bnu  阅读(393)  评论(0编辑  收藏  举报