单词统计难题

一 问题描述

Ignatius 最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

二 输入和输出

1 输入

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束。

2 输出

对于每个提问,给出以该字符串为前缀的单词的数量。

三 输入和输出样例

1 输入样例

banana

band

bee

absolute

acm

ba

b

band

abc

2 输出样例

2

3

1

0

四 分析和设计

1 分析

本问题为前缀统计问题。可以用字典树解决。

2 设计

a 将字符串依次插入到字典树中，插入过程中，用 cnt[p]  统计经过 P 节点的单词数。

b 在字典树中查询字符串，返回最后一个节点的 cnt[]，即为该字符串为前缀的单词数量。

五 代码

package com.platform.modules.alg.alglib.hdu1251;

public class Hdu1251 {
    private int maxn = 1005;
    private int maxz = 26; // 不同字符个数，例如数字 10，小写字母 26
    int trie[][] = new int[maxn][maxn];
    int cnt[] = new int[maxn]; // 计数器
    int tot; // 下标

    public String output = "";

    // 将字符串 s 插入到字典树中
    void insert(String s) {
        int len = s.length();
        int p = 1;
        for (int i = 0; i < len; i++) {
            int ch = s.charAt(i) - 'a'; // 转换成数字
            if (trie[p][ch] == 0)
                trie[p][ch] = ++tot;// 记录下标
            p = trie[p][ch];
            cnt[p]++; // 统计有多少个单词经过
        }
    }

    int query(String s) {
        int len = s.length(), p = 1;
        for (int i = 0; i < len; i++) {
            int ch = s.charAt(i) - 'a'; // 转换成数字
            p = trie[p][ch];
            if (p == 0)
                return 0;
        }
        return cnt[p];
    }

    public String cal(String input) {
        String s;
        tot = 1;
        String[] line = input.split("\n");
        int count = 0;
        while (true) {
            s = line[count++];
            if (s.length() == 0)
                break;
            insert(s);
        }
        while (true) {
            if (count == line.length)
                break;
            s = line[count++];
            if (s.length() == 0)
                break;
            output += query(s) + "\n";
        }
        return output;
    }
}

六 测试