【lintcode筆記】經典二分查找問題

在一個排序數組中找一個數,返回該數出現的任意位置,如果不存在,返回-1

樣例
給出數組 [1, 2, 2, 4, 5, 5].

對於 target = 2, 返回 1 或者 2.
對於 target = 5, 返回 4 或者 5.
對於 target = 6, 返回 -1.

//最基礎的二分查找算法
#include <stdio.h>
#include <vector>
#include <iostream>
using namespace std;
class Solution {
public:
    /**
     * @param A an integer array sorted in ascending order
     * @param target an integer
     * @return an integer
     */
    int findPosition(vector<int>& A, int target) {
        // Write your code here
        int n = A.size();
        if(n==0){
            return -1;
        }
        int start=0;
        int end=n-1;
        while(start+1 < end){  
            int mid = start+(end-start)/2;  
            if(A[mid] == target){  
                return mid;  
            }else if(A[mid] < target){  
                start = mid;  
            }else{  
                end = mid;  
            }  
        }  
        if(A[start] == target){  
            return start;  
        }  
        if(A[end] == target){  
            return end;  
        } 
        return -1;
    }
};

給定一個排序的整數數組(升序)和一個要查找的整數target,用O(logn)的時間查找到target 第一次出現的下標(從0開始),如果target不存在於數組中,返回-1。
樣例
在數組 [1, 2, 3, 3, 4, 5, 10] 中二分查找3,返回2。

加了一段往前搜索第一次出現的代碼。。。好像不太優雅

#include <stdio.h>
#include <vector>
#include <iostream>
using namespace std;
class Solution {
public:
    /**
     * @param nums: The integer array.
     * @param target: Target number to find.
     * @return: The first position of target. Position starts from 0. 
     */
    int binarySearch(vector<int> &array, int target) {
        // write your code here
        int n = array.size();
        if(n==0){
            return -1;
        }
        int start=0;
        int end=n-1;
        while(start+1 < end){  
            int mid = start+(end-start)/2;  
            if(array[mid] == target){
                while(array[mid] == array[mid-1]){
                    mid--;
                }
                return mid;  
            }else if(array[mid] < target){  
                start = mid;  
            }else{  
                end = mid;  
            }  
        }  
        if(array[start] == target){  
            return start;  
        }  
        if(array[end] == target){  
            return end;  
        } 
        return -1;
    }
};
發佈了30 篇原創文章 · 獲贊 19 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章