代码

#include "router.h"
#include <stdint.h>
#include <stdlib.h>
#include <map>
using namespace std;

map<uint32_t,RoutingTableEntry> table; //映射

/**
 * @brief 插入/删除一条路由表表项
 * @param insert 如果要插入则为 true ，要删除则为 false
 * @param entry 要插入/删除的表项
 *
 * 插入时如果已经存在一条 addr 和 len 都相同的表项，则替换掉原有的。
 * 删除时按照 addr 和 len **精确** 匹配。
 */
void update(bool insert, RoutingTableEntry entry) {
  // TODO:
   if(table.count(entry.addr))
    {
      RoutingTableEntry result=table[entry.addr];
      if(result.len==entry.len)
      {
        map<uint32_t,RoutingTableEntry>::iterator iter;
        iter = table.find(entry.addr);  
        table.erase(iter);
      }
    }
  if(insert)
  {
    table.insert(pair<uint32_t,RoutingTableEntry>(entry.addr,entry));
  }
}

/**
 * @brief 进行一次路由表的查询，按照最长前缀匹配原则
 * @param addr 需要查询的目标地址，网络字节序
 * @param nexthop 如果查询到目标，把表项的 nexthop 写入
 * @param if_index 如果查询到目标，把表项的 if_index 写入
 * @return 查到则返回 true ，没查到则返回 false
 */
bool prefix_query(uint32_t addr, uint32_t *nexthop, uint32_t *if_index) {
  // TODO:
  if(table.count(addr))
  {
    *nexthop=table[addr].nexthop;
    *if_index=table[addr].if_index;
    return true;
  }
  uint32_t addr2=addr%(0x1000000);
  if(table.count(addr2))
  {
    *nexthop=table[addr2].nexthop;
    *if_index=table[addr2].if_index;
    return true;
  }
  uint32_t addr3=addr%(0x10000);
  if(table.count(addr3))
  {
    *nexthop=table[addr3].nexthop;
    *if_index=table[addr3].if_index;
    return true;
  }
  uint32_t addr4=addr%(0x100);
  if(table.count(addr4))
  {
    *nexthop=table[addr4].nexthop;
    *if_index=table[addr4].if_index;
    return true;
  }
  if(table.count(0x00000000))
  {
    *nexthop=table[0x00000000].nexthop;
    *if_index=table[0x00000000].if_index;
    return true;
  }
  *nexthop = 0;
  *if_index = 0;
  return false;
}

评测结果