c++ stl模糊查找【找到key对应的值,找不到时返回最接近key的两个值】

在数据查找时如果找不到可以对应的值,有时候我们需要找距离key最近的值,可以利用upper_bound函数进行查找

如下两个函数GetMapInterVal和GetInterVal用于map版和其他线性容器

#include <map>

#include <vector>
#include <algorithm>
using namespace std;
//返回查找到的值(两个值相等)或最近的两个值,或者两个值都是起始或结尾点
template<typename T_Key,typename T_Val>
bool GetMapInterVal(const map<T_Key,T_Val> &_map,const T_Key &findKey,T_Val &preVal,T_Val &lastVal)
{
if (_map.size() == 0) return false;


auto iter = _map.find(findKey);
if (iter != _map.end())
{
preVal = iter->second;
lastVal = iter->second;
return true;
}


iter =  _map.upper_bound(findKey);
if (iter == _map.begin())
{
preVal = iter->second;
lastVal = iter->second;
}
else if (iter == _map.end())
{
--iter;
preVal = iter->second;
lastVal = iter->second;
}
else
{
lastVal = iter->second;
--iter;
preVal = iter->second;
}
return true;
}






//返回查找到的值(两个值相等)或最近的两个值,或者两个值都是起始或结尾点
//list/vector..
template<typename T, typename T_Val>
bool GetInterVal(const T &_arry,const T_Val &findVal,T_Val &preVal,T_Val &lastVal)
{
if (_arry.size() == 0) return false;


auto iter = find(_arry.begin(),_arry.end(),findVal);
if (iter != _arry.end())
{
preVal = *iter;
lastVal = *iter;
return true;
}




iter = upper_bound(_arry.begin(),_arry.end(),findVal);
if (iter == _arry.begin())
{
preVal = *iter;
lastVal = *iter;
}
else if (iter == _arry.end())
{
--iter;
preVal = *iter;
lastVal = *iter;
}
else
{
lastVal = *iter;
--iter;
preVal = *iter;
}


return true;

}

=============使用示例

vector<int>       _arry;
map<int,int>   _map;

for(int i = 0; i < 100;i+=5)
{
_map[i] = i;
_arry.push_back(i);
}

int FindKey = 12;
int preVal = -1;
int lastVal = -1;


//GetMapInterVal<int,int>(_map,FindKey,preVal,lastVal);
GetInterVal<vector<int>,int>(_arry,FindKey,preVal,lastVal);

返回10和15

发布了42 篇原创文章 · 获赞 7 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/jiyanglin/article/details/78936807