set源码之心得

C++的STL很强大，强大到我只愿慵懒地去使用而不知其所以然。直到李师问我，我的回答被李师否定，我方才意识到自己是多么地浅陋。希望自己有空抽时间把STL源码给研究一下，化为自己真正可以掌控的力量。

set容器的原型：template <class Key, class Compare=less<Key>, class Alloc=STL_DEFAULT_ALLOCATOR(Key) >。

在默认情况下，set容器使用less<key>进行比较。而我关心的就是这个less。平时我们会自定义一个比较函数来覆盖less（这里给个set基本用法的链接：http://www.cnblogs.com/jiu0821/p/4190026.html），里面往往是返回一个比较语句。传进去之后，set容器是怎么工作的呢？看过源码的coder会发现，工作原理极为简单：eg：

operator @;//传入的比较符号
type A, B;
if (A @ B) ...;
else if (B @ A) ...;
else ...
//得到A与B的“大小”关系

 

当下有这样一个需求：现有一个数据库，里面记录了若干IP段(即某个范围内所有IP)和IP点(某个IP)，现一个用户发来请求报文，我们需要判断该用户的IP是否在数据库中已记录。简化一下，这里的IP简化为一个正整数，大小不超过100。编码实现这个需求的解决方案。

思路：由于存在段和点，我们统一处理为段，即点转化为两端点相同的段，如此，我们求用户IP段与现有集合IP段是否有交集即可。设段为[t1,t2]，则利用set容器

if (a.t2 < b.t1) ...;
else if (b.t2 < a.t1) ...;
else ...;

即为所求。

代码：

#include <iostream>
#include <set>
using namespace std;

struct info
{
    int t1, t2;
    bool operator < (const info &r) const
    {
        return t2<r.t1;
    }
};

set<info> data;

void Input ();
bool Search ();
inline void Output (bool);

int main ()
{
    Input ();
    Output (Search ());
    return 0;
}

void Input ()
{
    info tmp;
    int m, n;
    cin >> m;//时间段
    for (int i = 1; i <= m; i++)
    {
        cin >> tmp.t1 >> tmp.t2;
        data.insert (tmp);
    }
    cin >> n;//时间点
    for (int i = 1; i <= n; i++)
    {
        cin >> tmp.t1;
        tmp.t2 = tmp.t1;
        data.insert (tmp);
    }
}
bool Search ()
{
    info tmp;
    set<info>::iterator it;
    cin >> tmp.t1;
    tmp.t2 = tmp.t1;
    it = data.find (tmp);
    if (it != data.end ())
    {
        return true;
    }
    else
    {
        return false;
    }
}
inline void Output (bool tmp)
{
    if (tmp)
    {
        cout << "succeed!"
             << endl;
    }
    else
    {
        cout << "failed!"
             << endl;
    }
}