逆序单词 HIhoCoder 1366 字典树

Posted 2020-12-01 alking1001

逆序单词 HIhoCoder 1366 字典序

题意

在英文中有很多逆序的单词，比如dog和god，evil和live等等。

现在给出一份包含N个单词的单词表，其中每个单词只出现一次，请你找出其中有多少对逆序单词。

第1行：1个整数，N，表示单词数量。2≤N≤50,000。

第2..N+1行：每行1个单词，只包含小写字母，每个单词长度不超过16个字母。保证每个单词只出现一次，且不会出现回文单词（即一个单词倒序还是它自己，比如eye）。

输出第一行是个整数，表示单词表中的逆序单词的对数。

解题思路

这个是我看B站学习字典序上看到的题，基本是个模板题。详情看代码实现吧。

代码实现

字典树大体有两种实现方式，区分是从结点的建立上，一种是使用new分配空间建立，另一种是预先开辟一个数组。

下面是两种形式的代码。第二种运行比较快。

#include<bits/stdc++.h> //时间是453ms
using namespace std;
struct Node
{
    Node *next[26]; //对应26个小写英文字母
    int flag; //这个记录从根节点到现在这个结点组成的单词是不是存在的单词，如果有多个的话，可以记录个数。
    Node()//构造函数
    {
        for(int i=0; i<26; i++)
            next[i]=NULL;
        flag=0;
    }
}*root;//这个root是字典树的第一个结点，也就是树的根。

void insert(char *s) //插入操作
{
    int len=strlen(s);
    Node *now = root;
    for(int i=len-1; i>=0; i--) //根据题目要求，这里使用逆序插入比较好。
    {
        int to=s[i]-'a';
        if(now->next[to]==NULL) //如果没有子节点，就建立。
            now->next[to]=new Node();
        now=now->next[to];//然后进入到下一个单词的子节点。
    }
    now->flag++; //可能有多个单词 
}
int fid(char *s)//查找函数
{
    int len=strlen(s);
    Node *now=root;
    for(int i=0; i<len; i++)//这里就是正向查找了
    {
        int to=s[i]-'a';
        if(now->next[to]==NULL)
            return 0;
        now=now->next[to];
    }
    return now->flag;//返回以这个结点为尾的单词的个数
} 
void del(Node *rot) //因为使用的是new分配的空间，所以使用完毕需要进行删除。
{
    for(int i=0; i<26; i++)
    {
        if(rot->next[i])
            del(rot->next[i]);//递归形式的删除。
    }
    delete(rot);
}
int main()
{
    int n, ans=0;
    char op[20];
    root = new Node();
    scanf("%d",&n);
    getchar();
    for(int i=0; i<n; i++)
    {
        scanf("%s", op);
        ans+=fid(op);
        insert(op);
    }
    printf("%d
", ans);
    del(root);
    return 0;
}

#include<bits/stdc++.h> //时间是242ms
using namespace std;
const int maxn=2e6+7;
int tree[maxn][27];
int flag[maxn];
int tot; 
void insert(char *str)
{
    int len=strlen(str);
    int root=0;
    for(int i=len-1; i>=0; i--)
    {
        int id=str[i]-'a';
        if(!tree[root][id])
            tree[root][id]=++tot; //类似于邻接链表的形式来建立
        root=tree[root][id];
    }
    flag[root]++;
}
int find(char *str) //需要根据需要进行匹配 
{
    int len=strlen(str);
    int root=0;
    for(int i=0; i<len; i++)
    {
        int id=str[i]-'a';
        if(!tree[root][id])
            return 0;
        root=tree[root][id];
    }
    return flag[root];
}
int main()
{
    int n, ans=0;
    char op[20];
    scanf("%d",&n);
    getchar();
    for(int i=0; i<n; i++)
    {
        scanf("%s", op);
        ans+=find(op);
        insert(op);
    }
    printf("%d
", ans);
    return 0;
}