c語言 二分法尋找 修正版 漏洞 二分法 32位

寫了一個二分法尋找的函數，主要避免有以下幾個容易出現漏洞的地方（適用於32位機器）。

 

1：unsigned int len 用unsigned int類型這樣可以尋找大於等於2G的資料

2：unsigned int la=0, he = len-1, mid; 用unsigned int類型 這樣在{mid = la+((he-la)>>1);}語句裡可以用位移，沒用{mid = (la+he)>>1;}是因為(la+he)可能會越界，即大於等於4G

3：用  if( mid == 0 ) return INVALID_LEN;是因為可能會出現mid == la == he == 0時, 再執行he = mid - 1; 後he=4g-1的情況

4: 返回unsigned int類型

 

 

#include <stdio.h>
#include <stdlib.h>

 

#define INVALID_LEN  0xfffffffff

 

unsigned int binsearch(int *p, unsigned int len, int key)
{
    if( p == NULL || len == 0 || len == INVALID_LEN ) return INVALID_LEN;
   
    unsigned int la=0, he = len-1, mid=0;
   
    while(la <= he )
    {
        mid = la+((he-la)>>1);
        if( key == p[ mid ] )
        {
            return  mid;
        }
        else if( key > p[ mid ] )
        {
            la = mid + 1;
        }
        else
        {
            if( mid == 0 ) return INVALID_LEN;
            he = mid - 1;
        }
    }
   
    return INVALID_LEN;
}                 
       
int main(int argc, char *argv[])
{
    const int len = 100;
   
    int *p = (int*)malloc(len*sizeof(int));
    if( p ==   NULL ) return -1;
   
    int i;
    for(i=0; i<len; i++) p[i] = i;
   
    printf("%d/n", binsearch(p, len, 0));
   
    system("PAUSE"); 
    return 0;
}