[leetcode] 34 find first and last position of element in sorted array

Given an array of integers nums sorted in ascending order, find the starting and ending position of a given target value.
Your algorithm's runtime complexity must be in the order of O(log n).
If the target is not found in the array, return [-1, -1].
Example 1:
Input: nums = [5,7,7,8,8,10], target = 8
Output: [3,4]
Example 2:
Input: nums = [5,7,7,8,8,10], target = 6
Output: [-1,-1]

有序数组中查找target 的 首次 和 最后一次出现下标

这题我用的方法很蠢, 就是 二分找出target, 然后O(n) 的往前,往后 找 first 和 last. 如果target有大量重复, 速度就会很慢, 但是这题 case 比较友好。。。

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        
        vector<int> ans;
        
        if(nums.empty()) return vector<int>{-1,-1};
        //step 1 : binary search target
        
        int begin = 0, end = nums.size()-1,mid = 0;
        
        while(begin +1 < end){
            mid = begin + (end - begin)/2;
            if(nums[mid] > target)
                end  = mid;
            else if(nums[mid] < target)
                begin = mid;
            else
                break;
        }
        if(nums[begin]!= target && nums[mid] != target && nums[end] != target )
            return vector<int>{-1,-1};
        
        // step 2 : find first position
        int initiate = 0;
        if(nums[begin] == target) 
            initiate = begin;
        else 
            initiate = nums[mid] == target? mid: end;
        // 越界情况
        while(initiate >=0 && nums[initiate]== target)
            initiate--;
        ans.push_back(++initiate);
        // step3: find last position
        
        //越界情况
         while(nums[initiate]== target && initiate <= nums.size()-1 )
             initiate++;
        ans.push_back(initiate-1);
        
        return ans; 
        
    }
};

image.png

我们显然不能面向case 编程, 该优化的还是要优化。
这里只需要对 binary search 做一些小的修改就好了

        while(begin+1 < end){
            mid = begin + (end- begin) /2;
            if(nums[mid] >= target)
                end = mid;
            else if(nums[mid] < target)
                begin = mid;       
        }

在find first position 的过程中,mid等于target 的时候, end 左移。
类似[1,2,3,3,3,3,3,3] --> [1,2,3,3,3]-->[1,2,3] 这样的过程

这样一般情况下,最后退出情况就是
begin < target, end= target,并且 begin, end 相邻。
比如[1,2,3,3,3,3,4] -->[2,3]

但是如果数组类似[2,2,2]的情况,begin 也有可能是first position
所以 我们这里 优先选择begin作为下标.

        end = nums[begin] == target? begin: end;
        ans.push_back(end );

find last postion 就是对[first element of target, ........, nums.size()-1]这个数组进行二分。

这是 数组内所有元素都大于等于target
所以 只要考虑将 begin 不断右移 ,
类似[2,2,2,2,2,3] --> [2,2,3]-->[2,3] 这种情况

使得最后退出条件是begin= target, end> target,begin,end相邻
同理,这里也要考虑[2,2,2]这种数组,

        begin = nums[end] == target? end:begin;
        ans.push_back(begin);

所以退出find last position 循环后, 优先选择end 作为 last element of target的下标。

完整代码如下。

class Solution {
public:
    vector<int> searchRange(vector<int>& nums, int target) {
        vector<int> ans;
        int begin = 0, end = nums.size()-1,mid= 0;
        
        // find first
        // 
        while(begin+1 < end){
            mid = begin + (end- begin) /2;
            if(nums[mid] >= target)
                end = mid;
            else if(nums[mid] < target)
                begin = mid;       
        }
        if(  nums.empty() || (nums[begin] != target && nums[end] != target) )
            return vector<int>{-1,-1};
        
        end = nums[begin] == target? begin: end;
        ans.push_back(end );
        
        // find last
        begin = end;
        end = nums.size()-1;
        while(begin +1<end){
            mid= begin + (end-begin)/2;
            if(nums[mid]> target)
                end = mid;
            else
                begin = mid;
        }
        
        begin = nums[end] == target? end:begin;
        ans.push_back(begin);
        
        return ans;
             
    }
};
image.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,126评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,254评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,445评论 0 341
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,185评论 1 278
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,178评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,970评论 1 284
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,276评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,927评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,400评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,883评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,997评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,646评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,213评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,204评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,423评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,423评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,722评论 2 345

推荐阅读更多精彩内容

  • rljs by sennchi Timeline of History Part One The Cognitiv...
    sennchi阅读 7,292评论 0 10
  • 2017.12.16 周六 多云 今天这太阳公公是露脸了,但温度也确实比昨天的雨天要稍低了点。...
    陈2468_TA76课程阅读 152评论 0 0