hdu1251

hdu1251

trie模板题,用g++会MLE,用c++就好了

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<set>
#include<map>
#include<stack>
#include<cstring>
#pragma GCC optimize(2)
#define inf 2147483647
#define ls rt<<1
#define rs rt<<1|1
#define lson ls,nl,mid,l,r
#define rson rs,mid+1,nr,l,r
#define N 1000010
#define For(i,a,b) for(int i=a;i<=b;++i)
#define p(a) putchar(a)
#define g() getchar()

using namespace std;
char c[20];
int t;
struct node{
    int cnt;
    node *next[26];
};
node *root=new node();
node *p;
void in(int &x){
    int y=1;char c=g();x=0;
    while(c<'0'||c>'9'){if(c=='-')y=-1;c=g();}
    while(c<='9'&&c>='0'){ x=(x<<1)+(x<<3)+c-'0';c=g();}
    x*=y;
}
void o(int x){
    if(x<0){p('-');x=-x;}
    if(x>9)o(x/10);
    p(x%10+'0');
}

void insert(){
    p=root;
    int l=strlen(c);
    For(i,0,l-1){
        t=c[i]-'a';
        if(p->next[t]==0)
            p->next[t]=new node();
        p=p->next[t];
        p->cnt++;
    }
}

int find(){
    p=root;
    int l=strlen(c);
    For(i,0,l-1){
        t=c[i]-'a';
        if(p->next[t]==0)
            return 0;
        p=p->next[t];
    }
    return p->cnt;
}

int main(){
    while(gets(c)&&c[0])
        insert();
    while(scanf("%s",c)!=EOF)
        o(find()),p('
');
    return 0;
}
原文地址:https://www.cnblogs.com/war1111/p/11251939.html