几道字典树题目

POJ 2418 Hardwood Species

题意：给一些字符串，按照字典序输出他们，并且输出频率...........

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一类的
#define MAX 100005
#define INF 0x7FFFFFFF
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
#define L(x) x<<1
#define R(x) x<<1|1
# define eps 1e-5
//#pragma comment(linker, "/STACK:36777216") ///传说中的外挂
using namespace std;

struct Trie {
    int next[128];
    int sum;
    char s[33];
    void init() {
        memset(next,0,sizeof(next));
        sum = 0;
    }
} a[301111];
int num,cnt,root;
char str[33];

void insert(char *key) {
    int p = root;
    for(int i=0; key[i]; i++) {
        int t = int(key[i]);
        if(a[p].next[t] == 0) {
            a[p].next[t] = ++ num;
            a[num].init();
        }
        p = a[p].next[t];
    }
    a[p].sum ++;
    a[p].s[0] = '\0';
    strcpy(a[p].s,key);
}

void dfs(int p) {
    if(a[p].sum != 0) {
        printf("%s ",a[p].s);
        printf("%.4f\n",(a[p].sum * 1.0) / cnt * 100);
    }
    for(int i=0; i<128; i++) {
        if(a[p].next[i] != 0) {
            int t = a[p].next[i];
            dfs(t);
        }
    }
}

int main() {
    root = 0; num = 0; cnt = 0;
    a[root].init();
    while(gets(str)) {
        int len = strlen(str);
        if(len == 0) break;
        cnt ++;
        insert(str);
    }
    dfs(0);
    return 0;
}

HDU 2846 Repository

题意：给定一些单词作为字典，和一些单词作为询问，对于每个询问，求出它在所有字典单词中作为单词子串出现的次数（它只能在同一单词中以子串形式出现一次）

对于每个字典中的单词，让它所有后缀都在字典树中建树，这样只要在字典树中找前缀就等同于找到子串了

需要注意的是，它只能在同一单词中以子串形式出现一次，如：str：dddddd       则ddd在str中只出现一次，所以对于同一单词产生的后缀，需要判重......

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一类的
#define MAX 100005
#define INF 0x7FFFFFFF
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
#define L(x) x<<1
#define R(x) x<<1|1
# define eps 1e-5
//#pragma comment(linker, "/STACK:36777216") ///传说中的外挂
using namespace std;
int n,q,root,cnt;
char str[22];
char keyword[22];

struct Trie {
    int next[26];
    int num;
    int kind;
    void init() {
        memset(next,0,sizeof(next));
        num = 0;
        kind = -1;
    }
} a[1111111];

void insert(char *key,int id,int kind) {
    int p = root;
    for(int i=id; key[i]; i++) {
        int t = key[i] - 'a';
        if(a[p].next[t] == 0) {
            a[p].next[t] = ++cnt;
            a[cnt].init();
        }
        p = a[p].next[t];
        if(a[p].kind != kind) {
            a[p].kind = kind;
            a[p].num ++;
        }
    }
}

int query(char *key) {
    int p = root;
    int sum = 0;
    for(int i=0; key[i]; i++) {
        int t = key[i] - 'a';
        if(a[p].next[t] == 0) return 0;
        p = a[p].next[t];
    }
    sum += a[p].num;
    return sum;
}

int main() {
    root = 0;
    cnt = 0;
    scanf("%d",&n);
    for(int i=0; i<n; i++) {
        scanf("%s",str);
        int len = strlen(str);
        for(int j=0; j<len; j++)
            insert(str,j,i);
    }
    scanf("%d",&q);
    for(int i=0; i<q; i++) {
        scanf("%s",keyword);
        printf("%d\n",query(keyword));
    }
    return 0;
}

POJ 1204 Word Puzzles

题意：给定了一个1000 * 1000 的由大写字母组成的矩阵，现在给出一些单词，要求在矩阵中找到每个单词，记录首字母的位置，以及它沿着什么方向走（8个方向，米字型）

将单词建立字典树，然后在矩阵中枚举每个点作为起点，8个方向直接找就行了..........

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <climits>//形如INT_MAX一类的
#define MAX 100005
#define INF 0x7FFFFFFF
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
#define L(x) x<<1
#define R(x) x<<1|1
# define eps 1e-5
//#pragma comment(linker, "/STACK:36777216") ///传说中的外挂
using namespace std;
int n,m,w;
struct Word {
    int x,y,dir,kind;
    char word[1111];
}s[1111];

char str[1111][1111];
int dx[] = {-1,-1,0,1,1,1,0,-1};
int dy[] = {0,1,1,1,0,-1,-1,-1};
int root , cnt ;
struct Trie {
    int next[26];
    int kind;
    void init() {
        memset(next,0,sizeof(next));
        kind = -1;
    }
}a[1111111];

void insert(char *key, int kind) {
    int p = root;
    for(int i=0; key[i]; i++) {
        int t = key[i] - 'A';
        if(a[p].next[t] == 0) {
            a[p].next[t] = ++ cnt;
            a[cnt].init();
        }
        p = a[p].next[t];
    }
    a[p].kind = kind;
}
void solve() {
    for(int i=0; i<n; i++) {
        for(int j=0; j<m; j++) {
            int t = str[i][j] - 'A';
            if(a[root].next[t] == 0) continue;
            for(int k=0; k<8; k++) {
                int p = a[root].next[t];
                int x = i, y = j;
                while(x >= 0 && x < n && y >= 0 && y < m && p) {
                    if(a[p].kind != -1) {
                        s[a[p].kind].dir = k;
                        s[a[p].kind].x = i;
                        s[a[p].kind].y = j;
                    }
                    x = x + dx[k];
                    y = y + dy[k];
                    int t = str[x][y] - 'A';
                    p = a[p].next[t];
                }
            }
        }
    }
    for(int i=0; i<w; i++) {
        printf("%d %d %c\n",s[i].x,s[i].y,s[i].dir + 'A');
    }
}

int main() {
    while(scanf("%d%d%d",&n,&m,&w) != EOF) {
        root = 0; cnt = 0;
        for(int i=0; i<n; i++) scanf("%s",str[i]);
        for(int i=0; i<w; i++) {
            scanf("%s",s[i].word);
            s[i].kind = i;
            insert(s[i].word,i);
        }
        solve();
    }
    return 0;
}

待续.....