代码

#include "router.h"
#include <arpa/inet.h>
#include <algorithm>
int nn;
RoutingTableEntry *a;
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);
        int lo=0,hi=nn;
        while (hi-lo>1){
            int mid=(lo+hi)/2;
            if (a[mid].addr>addr){
                hi=mid;
            }else{
                lo=mid;
            }
        }
        for (int j=hi-1;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;
}

评测结果