代码

#include <map>
#include <arpa/inet.h>
#include "router.h"

std::map<unsigned, unsigned> M[33];

void ins(const RoutingTableEntry *e) {
	M[e->len][htonl(e->addr)] = e->nexthop;
}

void init(int n, int q, const RoutingTableEntry *tbl) {
	for (int i = 0; i < n; i++) {
		ins(tbl + i);
	}
}

unsigned query(unsigned addr) {
	addr = htonl(addr);
	
	for (int i = 32; i >= 0; i--) {
		if (M[i].count(addr) > 0) {
			return M[i][addr];
		}
		addr &= ~(1u << (32 - i));
	}
	
	return 0;
}

评测结果