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 ;
 }

 void Input ()
 {
     info tmp;
     int m, n;
     cin >> m;//时间段
     for (int i = ; i <= m; i++)
     {
         cin >> tmp.t1 >> tmp.t2;
         data.insert (tmp);
     }
     cin >> n;//时间点
     for (int i = ; 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;
     }
 }