36 | AC自动机:如何用多模式串匹配实现敏感词过滤功能?

很多支持用户发表文本内容的网站,比如BBS,大都会有敏感词过滤功能,用来过滤掉用户输入的一些淫秽、反动、谩骂等内容。你有没有想过,这个功能是怎么实现的呢?

实际上,这些功能最基本的原理就是字符串匹配算法,也就是通过维护一个敏感词的字典,当用户输入一段文字内容之后,通过字符串匹配算法,来查找用户输入的这段文字,是否包含敏感词。如果有,就用“***”把它替代掉。

我们前面讲过好几种字符串匹配算法了,它们都可以处理这个问题。但是,对于访问量巨大的网站来说,比如淘宝,用户每天的评论数有几亿、甚至几十亿。这时候,我们对敏感词过滤系统的性能要求就要很高。毕竟,我们也不想,用户输入内容之后,要等几秒才能发送出去吧?我们也不想,为了这个功能耗费过多的机器吧?那如何才能实现一个高性能的敏感词过滤系统呢?这就要用到今天的多模式串匹配算法

基于单模式串和Trie树实现的敏感词过滤

我们前面几节讲了好几种字符串匹配算法,有BF算法、RK算法、BM算法、KMP算法,还有Trie树。前面四种算法都是单模式串匹配算法,只有Trie树是多模式串匹配算法。

我说过,单模式串匹配算法,是在一个模式串和一个主串之间进行匹配,也就是说,在一个主串中查找一个模式串。多模式串匹配算法,就是在多个模式串和一个主串之间做匹配,也就是说,在一个主串中查找多个模式串。

尽管,单模式串匹配算法也能完成多模式串的匹配工作。例如开篇的思考题,我们可以针对每个敏感词,通过单模式串匹配算法(比如KMP算法)与用户输入的文字内容进行匹配。但是,这样做的话,每个匹配过程都需要扫描一遍用户输入的内容。整个过程下来就要扫描很多遍用户输入的内容。如果敏感词很多,比如几千个,并且用户输入的内容很长,假如有上千个字符,那我们就需要扫描几千遍这样的输入内容。很显然,这种处理思路比较低效。

与单模式匹配算法相比,多模式匹配算法在这个问题的处理上就很高效了。它只需要扫描一遍主串,就能在主串中一次性查找多个模式串是否存在,从而大大提高匹配效率。我们知道,Trie树就是一种多模式串匹配算法。那如何用Trie树实现敏感词过滤功能呢?

我们可以对敏感词字典进行预处理,构建成Trie树结构。这个预处理的操作只需要做一次,如果敏感词字典动态更新了,比如删除、添加了一个敏感词,那我们只需要动态更新一下Trie树就可以了。

当用户输入一个文本内容后,我们把用户输入的内容作为主串,从第一个字符(假设是字符C)开始,在Trie树中匹配。当匹配到Trie树的叶子节点,或者中途遇到不匹配字符的时候,我们将主串的开始匹配位置后移一位,也就是从字符C的下一个字符开始,重新在Trie树中匹配。

基于Trie树的这种处理方法,有点类似单模式串匹配的BF算法。我们知道,单模式串匹配算法中,KMP算法对BF算法进行改进,引入了next数组,让匹配失败时,尽可能将模式串往后多滑动几位。借鉴单模式串的优化改进方法,能否对多模式串Trie树进行改进,进一步提高Trie树的效率呢?这就要用到AC自动机算法了。

经典的多模式串匹配算法:AC自动机

AC自动机算法,全称是Aho-Corasick算法。其实,Trie树跟AC自动机之间的关系,就像单串匹配中朴素的串匹配算法,跟KMP算法之间的关系一样,只不过前者针对的是多模式串而已。所以,AC自动机实际上就是在Trie树之上,加了类似KMP的next数组,只不过此处的next数组是构建在树上罢了。如果代码表示,就是下面这个样子:

public class AcNode {
  public char data; 
  public AcNode[] children = new AcNode[26]; // 字符集只包含a~z这26个字符
  public boolean isEndingChar = false; // 结尾字符为true
  public int length = -1; // 当isEndingChar=true时,记录模式串长度
  public AcNode fail; // 失败指针
  public AcNode(char data) {
    this.data = data;
  }
}

所以,AC自动机的构建,包含两个操作:

  • 将多个模式串构建成Trie树;

  • 在Trie树上构建失败指针(相当于KMP中的失效函数next数组)。

关于如何构建Trie树,我们上一节已经讲过了。所以,这里我们就重点看下,构建好Trie树之后,如何在它之上构建失败指针?

我用一个例子给你讲解。这里有4个模式串,分别是c,bc,bcd,abcd;主串是abcd。

Trie树中的每一个节点都有一个失败指针,它的作用和构建过程,跟KMP算法中的next数组极其相似。所以要想看懂这节内容,你要先理解KMP算法中next数组的构建过程。如果你还有点不清楚,建议你先回头去弄懂KMP算法。

假设我们沿Trie树走到p节点,也就是下图中的紫色节点,那p的失败指针就是从root走到紫色节点形成的字符串abc,跟所有模式串前缀匹配的最长可匹配后缀子串,就是箭头指的bc模式串。

这里的最长可匹配后缀子串,我稍微解释一下。字符串abc的后缀子串有两个bc,c,我们拿它们与其他模式串匹配,如果某个后缀子串可以匹配某个模式串的前缀,那我们就把这个后缀子串叫作可匹配后缀子串

我们从可匹配后缀子串中,找出最长的一个,就是刚刚讲到的最长可匹配后缀子串。我们将p节点的失败指针指向那个最长匹配后缀子串对应的模式串的前缀的最后一个节点,就是下图中箭头指向的节点。

计算每个节点的失败指针这个过程看起来有些复杂。其实,如果我们把树中相同深度的节点放到同一层,那么某个节点的失败指针只有可能出现在它所在层的上一层。

我们可以像KMP算法那样,当我们要求某个节点的失败指针的时候,我们通过已经求得的、深度更小的那些节点的失败指针来推导。也就是说,我们可以逐层依次来求解每个节点的失败指针。所以,失败指针的构建过程,是一个按层遍历树的过程。

首先root的失败指针为NULL,也就是指向自己。当我们已经求得某个节点p的失败指针之后,如何寻找它的子节点的失败指针呢?

我们假设节点p的失败指针指向节点q,我们看节点p的子节点pc对应的字符,是否也可以在节点q的子节点中找到。如果找到了节点q的一个子节点qc,对应的字符跟节点pc对应的字符相同,则将节点pc的失败指针指向节点qc。

如果节点q中没有子节点的字符等于节点pc包含的字符,则令q=q->fail(fail表示失败指针,这里有没有很像KMP算法里求next的过程?),继续上面的查找,直到q是root为止,如果还没有找到相同字符的子节点,就让节点pc的失败指针指向root。

我将构建失败指针的代码贴在这里,你可以对照着讲解一块看下,应该更容易理解。这里面,构建Trie树的代码我并没有贴出来,你可以参看上一节的代码,自己实现。

public void buildFailurePointer() {
  Queue<AcNode> queue = new LinkedList<>();
  root.fail = null;
  queue.add(root);
  while (!queue.isEmpty()) {
    AcNode p = queue.remove();
    for (int i = 0; i < 26; ++i) {
      AcNode pc = p.children[i];
      if (pc == null) continue;
      if (p == root) {
        pc.fail = root;
      } else {
        AcNode q = p.fail;
        while (q != null) {
          AcNode qc = q.children[pc.data - 'a'];
          if (qc != null) {
            pc.fail = qc;
            break;
          }
          q = q.fail;
        }
        if (q == null) {
          pc.fail = root;
        }
      }
      queue.add(pc);
    }
  }
}

通过按层来计算每个节点的子节点的失效指针,刚刚举的那个例子,最后构建完成之后的AC自动机就是下面这个样子:

AC自动机到此就构建完成了。我们现在来看下,如何在AC自动机上匹配主串?

我们还是拿之前的例子来讲解。在匹配过程中,主串从i=0开始,AC自动机从指针p=root开始,假设模式串是b,主串是a。

  • 如果p指向的节点有一个等于b[i]的子节点x,我们就更新p指向x,这个时候我们需要通过失败指针,检测一系列失败指针为结尾的路径是否是模式串。这一句不好理解,你可以结合代码看。处理完之后,我们将i加一,继续这两个过程;

  • 如果p指向的节点没有等于b[i]的子节点,那失败指针就派上用场了,我们让p=p->fail,然后继续这2个过程。

关于匹配的这部分,文字描述不如代码看得清楚,所以我把代码贴了出来,非常简短,并且添加了详细的注释,你可以对照着看下。这段代码输出的就是,在主串中每个可以匹配的模式串出现的位置。

public void match(char[] text) { // text是主串
  int n = text.length;
  AcNode p = root;
  for (int i = 0; i < n; ++i) {
    int idx = text[i] - 'a';
    while (p.children[idx] == null && p != root) {
      p = p.fail; // 失败指针发挥作用的地方
    }
    p = p.children[idx];
    if (p == null) p = root; // 如果没有匹配的,从root开始重新匹配
    AcNode tmp = p;
    while (tmp != root) { // 打印出可以匹配的模式串
      if (tmp.isEndingChar == true) {
        int pos = i-tmp.length+1;
        System.out.println("匹配起始下标" + pos + "; 长度" + tmp.length);
      }
      tmp = tmp.fail;
    }
  }
}

解答开篇

AC自动机的内容讲完了,关于开篇的问题,你应该能解答了吧?实际上,我上面贴出来的代码,已经是一个敏感词过滤的原型代码了。它可以找到所有敏感词出现的位置(在用户输入的文本中的起始下标)。你只需要稍加改造,再遍历一遍文本内容(主串),就可以将文本中的所有敏感词替换成“***”。

所以我这里着重讲一下,AC自动机实现的敏感词过滤系统,是否比单模式串匹配方法更高效呢?

首先,我们需要将敏感词构建成AC自动机,包括构建Trie树以及构建失败指针。

我们上一节讲过,Trie树构建的时间复杂度是O(m*len),其中len表示敏感词的平均长度,m表示敏感词的个数。那构建失败指针的时间复杂度是多少呢?我这里给出一个不是很紧确的上界。

假设Trie树中总的节点个数是k,每个节点构建失败指针的时候,(你可以看下代码)最耗时的环节是while循环中的q=q->fail,每运行一次这个语句,q指向节点的深度都会减少1,而树的高度最高也不会超过len,所以每个节点构建失败指针的时间复杂度是O(len)。整个失败指针的构建过程就是O(k*len)。

不过,AC自动机的构建过程都是预先处理好的,构建好之后,并不会频繁地更新,所以不会影响到敏感词过滤的运行效率。

我们再来看下,用AC自动机做匹配的时间复杂度是多少?

跟刚刚构建失败指针的分析类似,for循环依次遍历主串中的每个字符,for循环内部最耗时的部分也是while循环,而这一部分的时间复杂度也是O(len),所以总的匹配的时间复杂度就是O(n*len)。因为敏感词并不会很长,而且这个时间复杂度只是一个非常宽泛的上限,实际情况下,可能近似于O(n),所以AC自动机做敏感词过滤,性能非常高。

你可以会说,从时间复杂度上看,AC自动机匹配的效率跟Trie树一样啊。实际上,因为失效指针可能大部分情况下都指向root节点,所以绝大部分情况下,在AC自动机上做匹配的效率要远高于刚刚计算出的比较宽泛的时间复杂度。只有在极端情况下,如图所示,AC自动机的性能才会退化的跟Trie树一样。

内容小结

今天我们讲了多模式串匹配算法,AC自动机。单模式串匹配算法是为了快速在主串中查找一个模式串,而多模式串匹配算法是为了快速在主串中查找多个模式串。

AC自动机是基于Trie树的一种改进算法,它跟Trie树的关系,就像单模式串中,KMP算法与BF算法的关系一样。KMP算法中有一个非常关键的next数组,类比到AC自动机中就是失败指针。而且,AC自动机失败指针的构建过程,跟KMP算法中计算next数组极其相似。所以,要理解AC自动机,最好先掌握KMP算法,因为AC自动机其实就是KMP算法在多模式串上的改造。

整个AC自动机算法包含两个部分,第一部分是将多个模式串构建成AC自动机,第二部分是在AC自动机中匹配主串。第一部分又分为两个小的步骤,一个是将模式串构建成Trie树,另一个是在Trie树上构建失败指针。

课后思考

到此为止,字符串匹配算法我们全都讲完了,你能试着分析总结一下,各个字符串匹配算法的特点和比较适合的应用场景吗?

欢迎留言和我分享,也欢迎点击“请朋友读”,把今天的内容分享给你的好友,和他一起讨论、学习。

精选留言

  • zixuan

    2018-12-14 23:28:00

    思考题:
    一、单模式串匹配:
    1. BF: 简单场景,主串和模式串都不太长, O(m*n)
    2. KP:字符集范围不要太大且模式串不要太长, 否则hash值可能冲突,O(n)
    3. naive-BM:模式串最好不要太长(因为预处理较重),比如IDE编辑器里的查找场景; 预处理O(m*m), 匹配O(n), 实现较复杂,需要较多额外空间.
    4. KMP:适合所有场景,整体实现起来也比BM简单,O(n+m),仅需一个next数组的O(n)额外空间;但统计意义下似乎BM更快,原因不明.
    5. 另外查资料的时候还看到一种比BM/KMP更快,且实现+理解起来都更容易的的Sunday算法,有兴趣的可以看这里:
    http://www.inf.fh-flensburg.de/lang/algorithmen/pattern/sundayen.htm
    https://www.jianshu.com/p/2e6eb7386cd3

    二、多模式串匹配:
    1. naive-Trie: 适合多模式串公共前缀较多的匹配(O(n*k)) 或者 根据公共前缀进行查找(O(k))的场景,比如搜索框的自动补全提示.
    2. AC自动机: 适合大量文本中多模式串的精确匹配查找, 可以到O(n).
  • 西南偏北

    2018-12-14 11:26:14

    我只想说,老师你真牛X
  • 润鑫

    2018-12-14 14:52:00

    红黑树、KPM跟AC自动机这几节有点跟不上。。
  • O_o

    2018-12-17 15:03:17

    做安卓开发的,前边全部都理解+可动手手写。跟到最近几章感到面试可能确实用不到这些了,平时工作也确实用不到了。感谢老师最近的授课,通俗易懂!
    作者回复

    👍 厉害。最近这几讲不讲的话 知识就有缺陷 你可以不用太费劲看懂 知道有这个东西就行

    2018-12-18 09:55:12

  • bboy孙晨杰

    2018-12-19 17:11:43

    在看kmp和本节的ac自动机,很多文字描述我也理解不了,于是我就在纸上画一些具体的例子,然后按代码一步步的debug下去,虽然方法笨,但是很有助于理解。
  • 深蓝...

    2018-12-14 11:18:37

    完犊子了 从字符串匹配开始就掉队了 之前红黑树也是一脸懵逼。
  • zixuan

    2018-12-14 22:26:34

    前面激动说错了哈 ,跟DATrie没有半毛钱关系,后者只是一种Trie的具体实现.
    "其实,如果我们把树中相同深度的节点放到同一层,那么某个节点的失败指针只有可能出现在它所在层的上一层", 这里改成 "那么某个节点的失败指针只有可能指向比他所在层更小的层数的节点" 似乎更精确,虽然例子里刚好都是差一层,但实际应该可以往前跨多层的.
    和KMP算法一样,这个通过层次遍历来编织failNode数组的过程非常精妙,真的就像是织网一样。
  • coldpark

    2019-10-04 11:39:40

    fail数组的构建的作用我是这么理解的,请老师看看是不是对的:
    1. 在已经匹配上的敏感词中找到是否还有子集包含敏感词
    2.看这个子集的后续节点能否进一步匹配。
    举个例子:
    1. 敏感词是abc和bc,主串是abc,那么按照fail指针算法,abc中的c会链接到bc中的c,那么我匹配上了abc自然就相当于匹配上了bc,不用单独在主串中找是否含有bc。
    2. 主串是abcd,敏感词是abc,bcd,如果我匹配上abc,但是发现abc后面没有d,然后发现abc的c链接到bcd中的c,转过去一看,果然后面有d,就不用单独在主串中找是否含有bcd了。
  • roc

    2018-12-14 17:16:31

    王争老师,想问一下,我前面的内容掌握了有80%,如果不是面试算法岗,应该还算过关吧?
  • 张阔

    2020-01-13 17:03:42

    贴一个感觉不错的视频,可以结合着来看。https://www.bilibili.com/video/av81263689?p=1
  • skull

    2019-09-20 16:26:48

    https://www.cnblogs.com/sclbgw7/p/9260756.html,这篇文章跟老师写的文章互相补充着看,ac自动机的概念就一目了然了
  • blacknhole

    2018-12-23 15:55:36

    终于完全看懂了。
    有几个疑问:
    1,“首先 root 的失败指针为 NULL,也就是指向自己。”后半句是不准确或错误的,root的失败指针并非指向自身,因为root不等于null。
    2,“如果 p 指向的节点有一个等于 b[i] 的子节点 x……”以及下文中提到的b[i],是笔误吗?应该为a[i]吧,因为a才是主串。
  • 闫飞

    2019-01-17 08:14:34

    可以讲讲自动机的概念吧,否则总有些感觉突兀
    作者回复

    么机会了。专栏已经更新完了。不过,你的问题我记下来了,我会更新到我的公众号里,你可以关注我的公众号:“小争哥”

    2019-03-07 09:55:59

  • TryTs

    2018-12-19 15:42:12

    老师,我觉得学你这个课之后除了学习新的知识之外,还能够让我能够了解平时间那些常见应用背后的操作,最关键的时候在激发我的好奇心,让我能够去思考那些技术。嗯……我觉得很多时候好奇心就是学好知识的基础
  • EidLeung

    2018-12-14 17:38:55

    老师,如果要添加模式串,怎么改fail指针啊?
  • 懒猫

    2019-04-11 11:52:45

    老师,这里求最长可匹配后缀子串没理解,您举的例子:abc的最长可匹配后缀子串为bc,但是按照kmp的思想,abc的前缀子串为a、ab,后缀子串为c、bc,这里bc就不是最长可匹配后缀子串了呀,而且abc的最长可匹配后缀子串长度应该为0,不是吗
    作者回复

    你理解错了。这里说的最长可匹配后缀子串是:其他模式串可以匹配到abc的最长后缀子串。并不是abc自己的后缀子串匹配自己。

    2019-04-16 10:18:18

  • QQ怪

    2019-03-05 19:58:00

    正好要做这个敏感词过滤系统😂
  • 森鱼

    2019-09-04 22:05:00

    字符串这几节真烧脑……
    作者回复

    那就看看https://mp.weixin.qq.com/s/t8z4KQMrTrR3NljtWJm2zg

    2019-09-06 06:44:42

  • 文祥

    2019-03-20 20:11:39

    之前没看代码,一直在想到底怎么一层一层的给失败指针赋值,想破头也想不到。这一手linkedlist用也太巧妙了吧,保证了一层一层,从左到右给失败指针赋值,感动的我都哭了。
  • 佳娃

    2020-03-05 11:17:47

    知道有这个东西就行,以后遇到再来看!