C++学习之二分查找续

本文主要对上篇博文的 main函数 进行封装。

随机生成数据rand.cc 见上篇博文。

封装为函数及其各自的作用如下：

//读取数据到vec
void readfile(const string &filename , vector<int> &vec);
//二分查找
bool BinarySearch(const vector<int> &vec , int val);
//暴力查找
bool SimpleSearch(const vector<int> &vec , int val) ;
//查找数据 并打印出未在白名单里的数据以及这些数据的总和。 
int findandprint(const vector<int> &white , const vector<int> &data );
//程序运行时间
int64_t getTime();

 完整代码如下：

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <string.h>
#include <sys/time.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
using namespace std ;

//读取数据到vec
void readfile(const string &filename , vector<int> &vec);
//二分查找
bool BinarySearch(const vector<int> &vec , int val);
//暴力查找
bool SimpleSearch(const vector<int> &vec , int val) ;
//查找数据 并打印出未在白名单里的数据以及这些数据的总和。 
int findandprint(const vector<int> &white , const vector<int> &data );
//程序运行时间
int64_t getTime();


int main(int argc, const char *argv[])
{
    //读取white数据到文件
    //读取testcase 数据至文件
    //对whitecase排序
    //查找测试数据
    if(argc < 3)
    {
        fprintf(stderr , "Usage :%s , white ,testcase
", argv[0]);
        exit(EXIT_FAILURE);
    }
    int64_t  starttime = getTime();

    vector<int> white ;
    vector<int> testcase ;
    string whiteFile = argv[1];
    string testFile  = argv[2];

    readfile(whiteFile ,white);
    readfile(testFile , testcase);

    sort(white.begin() , white.end());

    int cnt = findandprint(white , testcase);

    int64_t  endtime = getTime();
    int64_t  cost = endtime - starttime ;

    cout << "白名单大小：" << white.size()
        << "测试数据大小：" << testcase.size()
        << "未出现在白名单的数据的总和：" << cnt
        << "花费时间："  << (cost/1000) << " ms" << endl ;
    return 0;
}

void readfile(const string &filename , vector<int> &vec)
{
    vec.clear();
    FILE* fp = fopen(filename.c_str() , "rb") ;
    if(NULL == fopen)
    {
        fprintf(stderr ,"Usage :%s open failure", filename.c_str());
        exit(EXIT_FAILURE);
    }

    char buf[128];
    while(memset(buf ,0,sizeof buf),fgets(buf ,sizeof buf ,fp)!= NULL)
    {
         int val = atoi(buf);
         vec.push_back(val);
    }
    fclose(fp);
}

bool BinarySearch(const vector<int> &vec , int val)
{
    int low = 0 ;
    int high = vec.size() - 1 ;

    while(low <= high)
    {
        int mid = (low + high)/ 2 ;
        if(val == vec[mid])
            return true ;
        else if(val > vec[mid])
            low = mid + 1 ;
        else
            high = mid - 1 ;
    }
    return false ;
}

bool SimpleSearch(const vector<int> &vec , int val)
{
    for(vector<int>::size_type ix = 0 ;ix != vec.size() ; ix++)
    {
        if(vec[ix] == val)
            return true ;
    }
    return false ;
}

int findandprint(const vector<int> &white , const vector<int> &data )
{
    int cnt = 0 ;
    for(vector<int>::size_type ix = 0 ; ix != data.size() ; ++ ix)
    {
        if(!BinarySearch(white , data[ix]))
        {
            cout << data[ix] << endl ;
            cnt ++ ;
        }
    }
    return cnt ;
}

int64_t getTime()
{
    struct timeval tm ;
    memset(&tm , 0 , sizeof(tm));
    if(gettimeofday(&tm ,NULL) == -1)
    {
        perror("gettimeofday");
    }
    
    int64_t t = 0 ;
    t += tm.tv_sec * 1000 * 1000;
    t += tm.tv_usec ;
    return t ;
}

本文编译代码：

g++ BinarySearch.cc

./a.out white.txt testcase.txt