【hihocoder 1627】Domains（字典树）

最新推荐文章于 2024-09-16 01:56:04 发布

weixin_33797791

最新推荐文章于 2024-09-16 01:56:04 发布

阅读量142

点赞数

文章标签：数据结构与算法

hihocoder 1627

The 2017 ACM-ICPC Asia Beijing Regional Contest 北京区域赛 A、Domains

题意

给N个VPN域名规则，由域名模式和VPN名字组成，域名模式中含有通配符，通配符*和#能匹配任何字符串，#只有在其他规则都不匹配时才生效。
合法的VPN域名模式：
1.只能含有通配符、小写字母、数字、点、连字符，
2.不能以点开头或结尾、两个点不能相连，
3.通配符必须在开头，后面跟着.或者空，
4.连字符必须在两个小写字母之间；
合法的VPN名字：
1.只能含有小写字母，数字和连字符，
2.以小写字母开头，
3.连字符必须在两个小写字母之间；
有的规则相互矛盾：
1.存在一个域名能同时匹配这两个规则，
2.因为一个规则的存在，另一个永远不会被匹配。
要求排除不合法的、和前面出现的规则矛盾的规则。
M个询问，每个询问给一个域名模式（保证合法且不含#）代表一个域名集合，求几个VPN规则能匹配该集合。

题解

排除不合法的域名。创建3棵字典树分别储存3种串（带*、带#、不带通配符），然后用字典树排除矛盾的情况。具体排除要讨论的细节太多了，再把合法的规则插入字典树。对询问分类讨论查询字典树。

代码

#include <bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,l,r) for(int i=l,ed=r;i<ed;++i)
#define per(i,l,r) for(int i=r-1,ed=l;i>=ed;--i)
#define all(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define pb push_back
#define mp make_pair
#define se second
#define sf(x) scanf("%d",&x)
#define db(x) cout<< #x <<"="<<(x)<<" "

typedef long long ll;
typedef double dd;
typedef vector<int> VI;
typedef pair<int,int> PII;
const ll LINF=0x3f3f3f3f3f3f3f3fLL;
const int INF=0x3f3f3f3f;
const dd EPS=1e-6;
const dd PI=acos(-1.0);

const ll mod=1e9+7;
const int N=451000;
const int MAX_N=40;
int n;
int get(char c){
    if(c>='a'&&c<='z')return c-'a';
    if(c>='0'&&c<='9')return c-'0'+26;
    if(c=='.')return 36;
    if(c=='-')return 37;
    if(c=='#')return 38;
    if(c=='*')return 39;
    return -1;
}
struct TrieNode{
    int cnt;
    int lf;
    int next[MAX_N];
};
struct Trie{
    TrieNode node[N];
    int tot;
    char o;
    Trie(char o=0):o(o){}
    void insert(string&s){
        int cur=0;
        ++node[cur].cnt;
        rep(i,0,SZ(s)){
            int p=get(s[i]);
            if(node[cur].next[p]==0)
                node[cur].next[p]=++tot;
            cur=node[cur].next[p];
            ++node[cur].cnt;
        }
        ++node[cur].lf;
    }
    void init(){
        tot=0;
        mem(node,0);
    }
    bool match(string&s,int jin=0){
        int cur=0;
        rep(i,0,SZ(s)){
            int p=get(s[i]);
            if(node[cur].next[p]==0){
                if(i){
                    if(s[i]=='*')return true;//遇到了*，且前面匹配
                    if(s[i]=='#'&&o=='#'||node[cur].next[get('#')]&&jin)return true;//遇到了#，另一个后面也是#，前面匹配
                }
                return (node[cur].next[get('*')]);
            }
            cur=node[cur].next[p];                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                             
        }
        return node[cur].lf;
    }
    int find(string&s,int xin,int &t){
        int cur=0;
        int ans=0;
        rep(i,0,SZ(s)){
            int p=get(s[i]);
            if(node[cur].next[p]==0){
                if(node[cur].next[get('*')]||node[cur].next[get('#')])return t=1;
                if(xin&&s[i]=='*')return ans+node[cur].cnt;
                return 0;
            }
            if(node[cur].next[get('*')])t=1;
            if(node[cur].next[get('#')])++ans;
            cur=node[cur].next[p];                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                       
        }
        return node[cur].lf;//都匹配了
    }
}xin('*'),jin('#'),non;
bool ckConflict(string s){
    int ch=s[0];
    reverse(s.begin(),s.end());
    if(ch=='#'){
        if(jin.match(s,1))return 0;
        if(xin.match(s,1))return 0;
        jin.insert(s);
        return 1;
    }else if(ch=='*'){
        if(non.match(s))return 0;
        if(jin.match(s))return 0;
        if(xin.match(s))return 0;
        xin.insert(s);
        return 1;
    }else {
        if(non.match(s))return 0;
        if(xin.match(s))return 0;
        non.insert(s);
        return 1;
    }
}
/*
a.c
poj.org vpn-china
www.pku.edu.cn vpn-pku1
mail.pku.edu.cn vpn-pku1
*.test.pku.edu.cn vpn-pku2
//*.test.pku.edu.cn vpn-pku2
//*.a.test.pku.edu.cn vpn-pku2
*.test2.pku.edu.cn vpn-pku2
//*.pku.edu.cn vpn-pku2
#.pku.edu.cn vpn-pku0
//#.www.pku.edu.cn vpn-pku0

*/
bool islower(char c){
    return c>='a'&&c<='z';
}
bool ckName(string&s){
    if(!s.size())return false;
    if(!islower(s[0]))return false;
    rep(i,1,SZ(s)){
        if(!(islower(s[i])||isalnum(s[i])||s[i]=='-'))
            return false;
        if(s[i]=='-'){
            if(!islower(s[i-1]))return false;
            if(i+1==SZ(s)||!islower(s[i+1]))return false;
        }
    }
    return true;
}

bool ckDomain(string&s){
    if(!SZ(s))return false;
    if(s[0]=='.')return false;
    if((s[0]=='#'||s[0]=='*')&&SZ(s)>1&&s[1]!='.')return false;
    rep(i,0,SZ(s)){
        int c=get(s[i]);
        if((c==-1)||(i&&c>37))return false;
        if(s[i]=='-'&&(!i||!islower(s[i-1])||i+1==SZ(s)||!islower(s[i+1])))return false;
        if(s[i]=='.'&&i+1<SZ(s)&&s[i+1]=='.')return false;
    }
    return true;
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("in","r",stdin);
#endif
    ios::sync_with_stdio(false);
    while(cin>>n){
        non.init();xin.init();jin.init();
        int VPN=0;
        rep(i,0,n){
            string name,domain;
            cin>>domain>>name;
            if(ckName(name)&&ckDomain(domain)&&ckConflict(domain)){
                ++VPN;
                //cout<<domain<<" "<<name<<endl;
            }
        }
        cout<<n-VPN<<endl;
        int q;
        cin>>q;
        rep(i,0,q){
            string s;
            int x=0,ans=0;
            cin>>s;
            if(s[0]=='*'){
                x=1;
                if(SZ(s)==1){
                    cout<<VPN<<endl;
                    continue;
                }
            }
            reverse(s.begin(),s.end());
            int b=0,nonFind;
            nonFind=non.find(s,x,b);
            ans=nonFind+xin.find(s,x,b);
            if(!b&&(x||!nonFind))ans+=jin.find(s,x,b);
            cout<<ans<<endl;
        }
    }
    return 0;
}