代码

#include "router.h"
#include <arpa/inet.h>
#include <algorithm>
int nn;
RoutingTableEntry *a;
bool addrcmp(const RoutingTableEntry &u, const RoutingTableEntry &v){
    return u.addr<v.addr;
}
void init(int n, int q, const RoutingTableEntry *tbl) {
    nn=n;a=(RoutingTableEntry *)tbl;
    for (int i=0;i<n;i++){
        a[i].addr=htonl(a[i].addr);
    }
}
unsigned query(unsigned addr) {
    unsigned addr0=addr;
    for (int i=32;i>=0;i--){
        addr=htonl(addr0);
        RoutingTableEntry tmp;
        tmp.addr=addr;
        int lo=std::lower_bound(a,a+nn,tmp,addrcmp)-a;
        for (int j=lo;j<nn;j++){
            if (a[j].addr==addr){
                if (a[j].len==i){
                    return a[j].nexthop;
                }
            }else{
                break;
            }
        }
        addr0&=~(1u<<((i-1)/8*8+7-(i-1)%8));
    }
    return 0;
}

评测结果