第六届山东省赛I题 Routing Table 字典树 NEU 1588 & UPC 3116 & SDUT 3259

最近一直没刷题 主要是快要网络考试了 一直在准备考试 当看到CIDR记法分配IP地址时 突然想到省赛这道题 如果懂点CIDR无分类编址的话 这道题还算比较好做 可惜当时并不懂 于是省赛上没做出来 主要思路就是把它给你的IP地址转化为二进制 保存在Trie树中 然后在网络前缀的最后一位保存IP地址的网络前缀位数和端口号 对于每一个目的IP地址 从第一位开始匹配 如果匹配 就更新信息 否则退出 一开始我是每一位都用vector保存这些信息 但是TLE MLE 很快我发现不用每一位都保存 因为比较网络前缀的前几位是没有意义的 因为就算前几位都匹配上了 只要你的网络前缀没有全部匹配上 就不能说这个目的IP地址属于这个网络 所以只需要在最后一位更新信息即可

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;

#define REP( i, a, b ) for( int i = a; i < b; i++ )
#define FOR( i, a, b ) for( int i = a; i <= b; i++ )
#define REV( i, a, b ) for( int i = a; i >= b; i-- )

const int maxn = 30 + 10;
const int INF = 1e9;

struct Trie{
          int bits;
          int port_nums;
          Trie *next[2];
          Trie(){
                    bits = -1;
                    port_nums = INF;
                    next[0] = NULL; next[1] = NULL;
          }
};

int n, m;

void Trans(int cur, int num, int *IP){
          int st = 8 * cur - 1;
          REV(i, st, st - 7){
                    if(num != 0){
                              IP[i] = num % 2;
                              num /= 2;
                    }
                    else IP[i] = 0;
          }
}

void CreatTrie(Trie *root, int *IP, int bit, int port_num){
          Trie *p = root;
          REP(i, 0, bit){
                    int id = IP[i];
                    if(p -> next[id] == NULL) p -> next[id] = new Trie;
                    p = p -> next[id];
          }
          //只在网络前缀的最后一位更新信息
          if(p -> bits < bit){
                    p -> bits = bit;
                    p -> port_nums = port_num;
          }
          else if(p -> bits == bit && p -> port_nums > port_num)
                    p -> port_nums = port_num;
}

void Build_Trie(Trie *root){
          int a[5], bit, port_num;
          int IP[maxn];
          REP(i, 0, n){
                    scanf("%d.%d.%d.%d/%d %d", &a[1], &a[2], &a[3], &a[4], &bit, &port_num);
                    FOR(j, 1, 4) Trans(j, a[j], IP);
                    CreatTrie(root, IP, bit, port_num);
          }
}

void FindTrie(Trie *root, int *IP, int &max_bits, int &port_num){
          Trie *p = root;
          REP(i, 0, 32){
                    int id = IP[i];
                    if(p -> next[id] == NULL) return;
                    p = p -> next[id];
                    if(p -> bits == max_bits && p -> port_nums < port_num)
                              port_num = p -> port_nums;
                    else if(p -> bits > max_bits){
                              port_num = p -> port_nums;
                              max_bits = p -> bits;
                    }
          }
}

void Check(Trie *root){
          int a[5], IP[maxn];
          REP(i, 0, m){
                    int max_bits = -1, port_num = INF;
                    scanf("%d.%d.%d.%d", &a[1], &a[2], &a[3], &a[4]);
                    FOR(j, 1, 4) Trans(j, a[j], IP);
                    FindTrie(root, IP, max_bits, port_num);
                    if(port_num != INF) printf("%d\n", port_num);
                    else printf("65535\n");
          }
}

void DelTrie(Trie *root){
          if(root == NULL) return;
          DelTrie(root -> next[0]);
          DelTrie(root -> next[1]);
          delete root;
}

void solve(){
          Trie *root = new Trie;
          scanf("%d%d", &n, &m);
          Build_Trie(root);
          Check(root);
          DelTrie(root);
}

int main()
{
          //freopen("in.txt", "r", stdin);
          int T;
          scanf("%d", &T);
          while(T--) solve();
          return 0;
}