过滤矢量并返回'n'元素

时间:2013-10-07 19:44:24

标签: c++ boost vector

我想编写一个懒惰求值的函数,并返回过滤后的向量的n元素。我大部分时间都对矢量中的第一个或第二个元素感兴趣。所以我不想过滤整个列表,然后找到n元素。

我正在学习使用Boost,如果有一个使用Boost的简单解决方案,那将非常有启发性。

int main() {

    double x[] = {10, 12.5, 12.9, 13.7, 50.07};
    size_t length = sizeof(x)/sizeof(x[0]);
    std::vector<double> vx(x, x+length);

    // Need a function to filter out elements less than 11 and return the 2nd
    // element greater than 11 (here 12.9) and return without evaluating 13.7 and 50.07


    return 0;
}

3 个答案:

答案 0 :(得分:3)

如果您对第一个元素感兴趣,我会天真地建议这样的事情:

std::vector<double>::iterator element(size_t n, const std::vector<double>& vec) {
    std::vector<double>::iterator it = vec.begin();
    size_t found = 0;
    while (found != n && it != vec.end()) {
        if (*(it++) >= 11) ++found;
    }
    return it;
}

我有一个线性复杂性,但只要找到所需的匹配就会退出。

答案 1 :(得分:1)

使用std::partition

这样的事情
float n=11.f;
auto it =std::partition(vx.begin(), vx.end(), 
                       [n](const double & p){ return p <n;});

 it++; //Second element
 if( it!= vx.end())
    std::cout<<"Element :"<<*it<<std::endl;

请参阅here

答案 2 :(得分:0)

我不知道如何使用提升但这是我会使用的:

int binarySearch(int whatToSearch){
    //recursive binary search 
    if value is found 
        return index
    else //not found
        return -index-1; //-index+1 will give you the place to insert that element    
}

然后我会在返回索引后得到第二个元素(如果存在)。

以下是我使用的二进制搜索代码

int mybinary_search(string array[],int first,int last, string search_key){
    int index;

    if (first > last)
        index = -first-1;

    else{
        int mid = (first + last)/2;

        if (search_key == array[mid])
            return mid;
        else if (search_key < array[mid])
        index = mybinary_search(array,first, mid-1, search_key);
        else
            index = mybinary_search(array, mid+1, last, search_key);

    } // end if
    return index;
}

double findAskedValue(int value){
    double x[] = {10, 12.5, 12.9, 13.7, 50.07};
    size_t length = sizeof(x)/sizeof(x[0]);

    int index = mybinarySearch(x,0,length, 11);
    if(index >= 0)
         cout << "value I'm searching for " << x[index+2] << endl;
    else
         cout << "value I'm searching for " << x[(-index-1)+2] << endl;
     //instead of cout's you can do what ever you want using that index
}

一个小解释:你打算调用findAskedValue(val),它会返回一个值。 如果值&gt; 0,然后元素存在于列表中,index + 2是您要搜索的值的位置,否则(-index-1)是插入该元素的位置。 -index-1 + 2将为您提供第二个更大元素。

复杂性为O(log(n))