hdu 4760 Good Firewall(字典树)

本文主要是介绍hdu 4760 Good Firewall(字典树)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

题目链接：hdu 4760 Good Firewall

题目大意：有一个防火墙，具有添加一个子网络，删除一个子网络，以及转发包的操作。

添加操作包含子网络的id，以及子网络的子网掩码（计算出网络前缀，以及ip的下限），不会超过15个。
删除则是给定要删除的子网络id。
转发操作，给定两个ip，如果两个ip在同一个子网络中，则可以转发，否则丢弃。

解题思路：对子网掩码前缀建立字典树，每个前缀终止节点用一个set记录属于哪些子网络，ip下限。那么增加和删除操

作既可以解决了。对于查询操作，分别查询两个ip，处理除两个ip可能属于的网络，判断有无共同即可。

#include <cstdio>
#include <cstring>
#include <set>
#include <algorithm>using namespace std;typedef long long ll;
typedef pair<int, ll> pii;
typedef set<pii>::iterator iter;
const int maxn = 1024 * 15 + 10;
const int maxm = 105;
const int sigma_size = 2;struct Tire {int sz, sv;int g[maxn * maxm][sigma_size];int idx[maxn * maxm], cnt[1030];set<pii> ans, vis[maxn];void init();int newSet();void addSet(int id, ll limt);void insert(char* str, pii x);void remove(char* str, pii x);void find(char* str);bool judge(char* a, char* b);
}T;struct Network {int n;ll suf[20];char ip[20][maxm];Network() { n = 0; memset(suf, 0, sizeof(suf));}
}net[1030];ll change(char* ans, bool flag) {char str[105];int n = strlen(str), a[4], b;if (flag)scanf("%d.%d.%d.%d/%d", &a[0], &a[1], &a[2], &a[3], &b);else {scanf("%d.%d.%d.%d", &a[0], &a[1], &a[2], &a[3]);b = 32;}int t = 0;ll ret = 0;for (int i = 0; i < 4; i++) {for (int j = 7; j >= 0; j--) {if (t < b)ans[t] = ((a[i]>>j)&1) + '0';else if (((a[i]>>j)&1))ret |= (1LL<<(31-t));t++;}}ans[t] = '\0';return ret;
}int main () {int id;char op[5], a[maxm], b[maxm], c[maxm];T.init();while (scanf("%s", op) == 1) {if (op[0] == 'E') {scanf("%d", &id);scanf("%d", &net[id].n);for (int i = 0; i < net[id].n; i++) {net[id].suf[i] = change(net[id].ip[i], a);T.insert(net[id].ip[i], make_pair(id, net[id].suf[i]));}} else if (op[0] == 'D') {scanf("%d", &id);for (int i = 0; i < net[id].n; i++)T.remove(net[id].ip[i], make_pair(id, net[id].suf[i]));net[id].n = 0;} else {change(a, 0);change(b, 0);printf("%c\n", T.judge(a, b) ? 'F' : 'D');}}return 0;
}bool Tire::judge(char* a, char* b) {memset(cnt, 0, sizeof(cnt));T.find(a);for (iter i = ans.begin(); i != ans.end(); i++)cnt[i->first] = 1;T.find(b);for (iter i = ans.begin(); i != ans.end(); i++)if (cnt[i->first])return true;return false;
}void Tire::init() {sz = sv = 1;idx[0] = 0;vis[0].clear();memset(g[0], 0, sizeof(g[0]));
}int Tire::newSet() {vis[sv].clear();return sv++;
}void Tire::addSet(int id, ll limt) {for (iter i = vis[id].begin(); i != vis[id].end(); i++)if (i->second <= limt)ans.insert(*i);
}void Tire::insert(char* str, pii x) {int u = 0, n = strlen(str);for (int i = 0; i < n; i++) {int v = str[i] - '0';if (g[u][v] == 0) {idx[sz] = 0;memset(g[sz], 0, sizeof(g[sz]));g[u][v] = sz++;}u = g[u][v];}if (idx[u] == 0)idx[u] = newSet();vis[idx[u]].insert(x);
}void Tire::remove(char* str, pii x) {int u = 0, n = strlen(str);for (int i = 0; i < n; i++) {int v = str[i] - '0';if (g[u][v] == 0) {idx[sz] = 0;memset(g[sz], 0, sizeof(g[sz]));g[u][v] = sz++;}u = g[u][v];}vis[idx[u]].erase(x);
}void Tire::find(char* str) {ans.clear();ll ret = 0;int u = 0, n = strlen(str);for (int i = 0; i < n; i++) {int v = str[i] - '0';if (g[u][v] == 0)return;u = g[u][v];if (idx[u]) {ll ret = 0;for (int j = i+1; j < n; j++)if (str[j] == '1')ret |= (1LL<<(31-j));addSet(idx[u], ret);}}
}