AcWing 835. Trie字符串统计

最新推荐文章于 2022-12-14 11:11:33 发布

原创最新推荐文章于 2022-12-14 11:11:33 发布 · 275 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#java #数据结构 #字符串

AcWing算法基础专栏收录该内容

78 篇文章

订阅专栏

本文介绍了一种维护字符串集合的方法，支持插入和查询操作。通过构建串树来存储字符串，并使用数组记录每个字符串出现的次数。文章提供了完整的Java代码实现，包括插入和查询功能，适用于处理大量字符串操作场景。

维护一个字符串集合，支持两种操作：

“I x”向集合中插入一个字符串x；
“Q x”询问一个字符串在集合中出现了多少次。

共有N个操作，输入的字符串总长度不超过 105105，字符串仅包含小写英文字母。

输入格式

第一行包含整数N，表示操作数。

接下来N行，每行包含一个操作指令，指令为”I x”或”Q x”中的一种。

输出格式

对于每个询问指令”Q x”，都要输出一个整数作为结果，表示x在集合中出现的次数。

每个结果占一行。

数据范围

1≤N≤2∗1041≤N≤2∗104

输入样例：

5
I abc
Q abc
Q ab
I ab
Q ab

输出样例：

1
0
1

import java.io.*;
import java.lang.Integer;

class Main{
    static int N = 100010;
    static int[][] t = new int[N][26];//串树，保存26中字母
    static int[] cnt = new int[N];//保存单词出现的个数
    static int idx = 0;//链接字符与字符的下标
    
    static void insert(String str){
        int p = 0;
        for(int i = 0; i < str.length(); ++i){
            int c = str.charAt(i) - 'a';
            if(t[p][c] == 0){//表示没有c下标对应的字符
                t[p][c] = ++idx;//记录当前出现字符的位置，以便于往下继续插入字符
            }
            p = t[p][c];//p记录的是下一个字符的坐标
        }
        cnt[p]++;//字符串插入完毕，记录当前字符串插入的次数
    }
    //查找过程和插入过程很相似，是遍历的过程
    static int query(String str){
        int p = 0;
        for(int i = 0;i < str.length(); ++i){
            int c= str.charAt(i) - 'a';
            if(t[p][c] == 0)return 0;
            p = t[p][c];
        }
        return cnt[p];
    }
    public static void main(String[] args)throws Exception{
        BufferedReader buf = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter buw = new BufferedWriter(new OutputStreamWriter(System.out));
        int n = Integer.valueOf(buf.readLine());
        for(int i = 0; i < n; ++i){
            String[] params = buf.readLine().split(" ");
            if("I".equals(params[0])){
                insert(params[1]);
            }else{
                buw.write(query(params[1]) + "\n");
            }
        }
        buw.flush();
        buf.close();
        buw.close();
    }
}