剑指 Offer II 006. 排序数组中两个数字之和

水题,vis记录,暴力即可

注意有重复

class Solution {
public:
    vector<int> twoSum(vector<int>& numbers, int target) {
        int len = numbers.size();
        int vis[3001];
        memset(vis, 0, sizeof(vis));
        for(int i = 0; i < len; i++)
        {
            if(numbers[i] < 0)
            {
                int temp = abs(numbers[i]);
                if(vis[temp] == 0) vis[temp] = 2;
                else if(vis[temp] == 1) vis[temp] = 3;
                else if(vis[temp] == 2) vis[temp] = 4;
            }
            else
            {
                int temp = numbers[i];
                if(vis[temp] == 0) vis[temp] = 1;
                else if(vis[temp] == 2) vis[temp] = 3;
                else if(vis[temp] == 1) vis[temp] = 5;
            }
        }
        vector<int> ret;
        int flag = 0, k;
        for(int i = 0; i < len; i++)
        {
            int d = target - numbers[i];
            if(d == numbers[i])
            {
                if(d < 0 && vis[-d] == 4) ret.push_back(i);
                if(d >= 0 && vis[d] == 5) ret.push_back(i);
            }
            else if(d < 0)
            {
                d = -d;
                if(vis[d] == 2 || vis[d] == 3)
                {
                    ret.push_back(i);
                }
            }
            else
            {
                if(vis[d] == 1 || vis[d] == 3)
                {
                    ret.push_back(i);
                }
            }

        }
        return ret;

    }
};

 

posted @ 2022-03-29 00:07  WTSRUVF  阅读(16)  评论(0编辑  收藏  举报