首页 > 代码库 > 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);
        }
    }
}

hdu 4760 Good Firewall(字典树)