由于笔者公司得前端需求缘故,笔者写了一个关于js版得ac自动机算法,可以修改对应得关键词组,如果只用到一组,就删除其余无关方法及集合等。代码如下:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<script type="application/javascript">
class TrieNode {
constructor() {
this.sunChildren = new Map(); // 子节点映射表
this.mainChildren = new Map(); // 主节点映射表
this.subChildren = new Map(); // 副节点映射表
this.isSunEndOfWord = false; // 子是否是关键词结尾
this.isMainEndOfWord = false; // 主是否是关键词结尾
this.isSubEndOfWord = false; // 副是否是关键词结尾
this.sunFail = this; // 子失败指针
this.mainFail = this; // 主失败指针
this.subFail = this; // 副失败指针
this.sunKeywords = new Set(); // 子存储以该节点结尾的关键词
this.mainKeywords = new Set(); // 主存储以该节点结尾的关键词
this.subKeywords = new Set(); // 副存储以该节点结尾的关键词
}
}
class Trie {
constructor() {
this.root = new TrieNode();
}
insertMainKeywords(mainKeywords) {
if(mainKeywords.length===0){
return
}
let current = this.root;
for (const keyword of mainKeywords) {
for (const ch of keyword) {
current.mainChildren.set(ch, new TrieNode())
current = current.mainChildren.get(ch);
}
current.isMainEndOfWord = true;
current.mainKeywords.add(keyword);
current = this.root
}
}
buildMainFailPointers() {
let queue = [] ;
for (let child of this.root.mainChildren.values()) {
child.mainFail = this.root
queue.push(child)
}
while (queue.length !== 0) {
const current = queue.shift();
current.mainChildren.forEach((value, key) => {
let ch = key
let child = value;
let failNode = current.mainFail;
while (!Object.keys(failNode) && !failNode.mainChildren.has(ch)){
failNode = failNode.mainFail
}
if (Object.keys(failNode)){
child.mainFail = this.root
}else {
child.mainFail = failNode.mainChildren.get(ch);
if (child.mainKeywords.size!==0) {
child.mainKeywords.add(child.mainFail.mainKeywords);
}
}
queue.push(child);
})
}
}
searchMainKeywords(text) {
if(text.length===0){
return
}
let matchedKeywords = new Set()
let current = this.root;
let matchedMainKeyword = new Array(text.length)
let index = 0
for (let ch of text) {
let children = current.mainChildren
while (current!==this.root && !children.has(ch)) {
current = current.mainFail
children = current.mainChildren
index = 0;
}
if (children.has(ch)) {
current = children.get(ch)
matchedMainKeyword[index++]=ch;
if (current.isMainEndOfWord){
matchedKeywords.add(matchedMainKeyword.join(""))
}
}else {
current = this.root;
index = 0;
}
}
return matchedKeywords;
}
insertSubKeywords(mainKeywords) {
if(mainKeywords.length===0){
return
}
let current = this.root;
for (const keyword of mainKeywords) {
for (const ch of keyword) {
current.subChildren.set(ch, new TrieNode())
current = current.subChildren.get(ch);
}
current.isSubEndOfWord = true;
current.sunKeywords.add(keyword);
current = this.root
}
}
buildSubFailPointers() {
let queue = [] ;
for (let child of this.root.subChildren.values()) {
child.subFail = this.root
queue.push(child)
}
while (queue.length !== 0) {
const current = queue.shift();
current.subChildren.forEach((value, key) => {
let ch = key
let child = value;
let failNode = current.subFail;
while (!Object.keys(failNode) && !failNode.subChildren.has(ch)){
failNode = failNode.subFail
}
if (Object.keys(failNode)){
child.subFail = this.root
}else {
child.subFail = failNode.subChildren.get(ch);
if (child.subKeywords.size!==0) {
child.subKeywords.add(child.subFail.subKeywords);
}
}
queue.push(child);
})
}
}
searchSubKeywords(text) {
if(text.length===0){
return
}
let matchedKeywords = new Set()
let current = this.root;
let matchedMainKeyword = new Array(text.length)
let index = 0
for (let ch of text) {
let children = current.subChildren
while (current!==this.root && !children.has(ch)) {
current = current.subFail
children = current.subChildren
index = 0;
}
if (children.has(ch)) {
current = children.get(ch)
matchedMainKeyword[index++]=ch;
if (current.isSubEndOfWord){
matchedKeywords.add(matchedMainKeyword.join(""))
}
}else {
current = this.root;
index = 0;
}
}
return matchedKeywords;
}
insertSunKeywords(mainKeywords) {
if(mainKeywords.length===0){
return
}
let current = this.root;
for (let keyword of mainKeywords) {
for (const ch of keyword) {
current.sunChildren.set(ch, new TrieNode())
current = current.sunChildren.get(ch);
}
current.isSunEndOfWord = true;
current.sunKeywords.add(keyword);
current = this.root
}
}
buildSunFailPointers() {
let queue = [] ;
for (let child of this.root.sunChildren.values()) {
child.sunFail = this.root
queue.push(child)
}
while (queue.length !== 0) {
const current = queue.shift();
current.sunChildren.forEach((value, key) => {
let ch = key
let child = value;
let failNode = current.sunFail;
while (!Object.keys(failNode) && !failNode.sunChildren.has(ch)){
failNode = failNode.sunFail
}
if (Object.keys(failNode)){
child.sunFail = this.root
}else {
child.sunFail = failNode.sunChildren.get(ch);
if (child.sunKeywords.size!==0) {
child.sunKeywords.add(child.subFail.sunKeywords);
}
}
queue.push(child);
})
}
}
searchSunKeywords(text) {
if(text.length===0){
return
}
let matchedKeywords = new Set()
let current = this.root;
let matchedMainKeyword = new Array(text.length)
let index = 0
for (let ch of text) {
let children = current.sunChildren
while (current!==this.root && !children.has(ch)) {
current = current.sunFail
children = current.sunChildren
index = 0;
}
if (children.has(ch)) {
current = children.get(ch)
matchedMainKeyword[index++]=ch;
if (current.isSunEndOfWord){
matchedKeywords.add(matchedMainKeyword.join(""))
}
}else {
current = this.root;
index = 0;
}
}
return matchedKeywords;
}
}
let startTime=new Date().getTime();
// 使用示例
const trie = new Trie();
trie.insertMainKeywords(["需求","关注","控制","养成"]);
trie.buildMainFailPointers();
trie.insertSubKeywords(["需求","关注","控制","养成"]);
trie.buildSubFailPointers();
trie.insertSunKeywords(["需求","关注","控制","养成"]);
trie.buildSunFailPointers();
const text = "但这似乎又确实证明了一点,郑州大学一旦出了省,名气是真的惨(😭),这上哪说理去。\n" +
"\n" +
"但我肯定不会无脑地劝大家梭哈计算机,就像当年同桌劝我梭哈机械自动化一样。现在是信息化时代,网络上铺天盖地的消息大家都能看得到。\n" +
"\n" +
"自己适合什么专业,喜欢什么专业,分数能上什么大学,其实在上大学之前确实是应该充分考量一下的。但个人意志在时代的洪流之下,往往显得微不足道。\n" +
"\n" +
"就像我当年虽然报了机械自动化,其他志愿里也压根没有选和计算机相关的专业,但因为分数的问题,竟然被调剂到了计算机网络专业。\n" +
"\n" +
"曾经天真的以为这就是学校最垃圾的专业,就是单纯地为了扩招应付我们这群高考失意的人,结果没想到,赶上了计算机专业的黄金时代。\n" +
"\n" +
"这种命运的捉弄,除了感慨“时也命也”我实在想不出来更好的解释。\n" +
"\n" +
"我算是一个很努力的人了,在技术公众号里是少有的能每天坚持原创的博主,但我必须得承认,如果我当年没有调剂到计算机专业,没有学 Java,没有大三就去一家世界五百强的外企实习,没有从 2014 年就开始坚持写博客,我现在注定是籍籍无名的、默默无闻的。\n" +
"\n" +
"都说现在很卷,但有一说一,我们园区一到 6 点基本上人都走光了,晚上也很少有人加班;但这并不代表别的园区就不加班。\n" +
"\n" +
"都说计算机专业很卷,但有一说一,随着亟需就业的人数越来越多,生产力资料受限的情况下,其他的专业也一样面临着同样的问题。\n" +
"\n" +
"所以,选择学校和专业有时候是一个很随机的事情,没准你现在选了火热的计算机专业,50 年后真的就毫无用处;没准你现在选了一个冷门专业,比如说考古专业,没准 50 年后真的人手一把洛阳铲。\n" +
"\n" +
"回到一个实际的问题,“择校时应该看重专业还是学校呢?”\n" +
"\n" +
"我的结论是优先学校。\n" +
"\n" +
"简历命名的时候,我一般推荐的就是姓名-学校-硕士(985、211).pdf,如果这些都不是,那命名就是姓名-学校-简历.pdf,大家可以细品再细品,含金量可想而知。\n" +
"\n" +
"反而是专业,显得就没有那么重要了,没有人会写姓名-学校-计算机科学与技术专业.pdf。\n" +
"\n" +
"科班和非科班,在计算机就业方面的影响,其实很微弱。如果你非要反驳,我也是不打算反驳你的(😁)。在当今社会,人们越来越关注身体健康和饮食安全。然而,随着生活节奏的加快和竞争压力的增大,越来越多的人开始忽视自己的饮食健康。为了改善这种状况,本文将探讨如何制定一个健康的饮食计划,以达到保持身体健康的目的。\n" +
"\n" +
"在制定健康饮食计划之前,我们需要了解一些基本原则。首先,要保持饮食的平衡,包括各种营养素的如蛋白质、脂肪、碳水化合物、维生素和矿物质的摄入。其次,要控制摄入的总能量,避免过度摄入导致肥胖和相关疾病。最后,要注意食物的质量和来源,选择新鲜、无污染的食物,减少摄入过多的添加剂和农药残留。\n" +
"\n" +
"为了实现饮食平衡和控制能量摄入,我们可以采取以下措施。首先,合理搭配五大类食物:谷物、蔬菜、水果、肉类、豆类,确保摄入全面均衡的营养。其次,适量摄入优质蛋白质,如鱼、鸡胸肉、豆腐等,以满足身体需要。第三,控制脂肪摄入,尤其是饱和脂肪和反式脂肪的摄入,以预防心血管疾病。最后,注意控制糖分和盐分的摄入,避免引发糖尿病和高血压等疾病。\n" +
"\n" +
"除了合理搭配食物,正确的饮食习惯也是保持健康的关键。首先,要养成定时定量饮食的习惯,避免暴饮暴食。其次,要慢慢咀嚼食物,有助于消化吸收。此外,适当运动可以促进消化和吸收,增强身体免疫力。\n" +
"\n" +
"在制定健康饮食计划时,还需要注意一些特殊人群的饮食需求。例如,孕妇需要额外补充叶酸等营养素;儿童需要多摄入钙、铁等营养素;老年人则需要控制碳水化合物和脂肪的摄入。\n" +
"\n" +
"总之,制定一个健康的饮食计划是保持身体健康的重要手段。通过合理搭配食物、控制能量摄入、养成良好饮食习惯以及关注特殊人群的饮食需求,我们可以实现饮食平衡和身体健康。让我们从今天开始,关注饮食健康,享受美好生活!";
const matchedKeywords = trie.searchMainKeywords(text);
const matchedKeywords1 = trie.searchSubKeywords(text);
const matchedKeywords2 = trie.searchSunKeywords(text);
// trie.searchPairsIds(text,["军衔", "审查", "条例", "中国"])
// console.log(matchedKeywords,new Date().getTime()-startTime+"ms"); // 应输出: ["他们", "们", "人"]
console.log(matchedKeywords,new Date().getTime()-startTime+"ms"); // 应输出: ["他们", "们", "人"]
let startTime1=new Date().getTime();
let res = ["需求","关注","控制","养成"]
res.forEach(v=>{
let b = text.includes(v)
if (b){
console.log(v)
}
})
res.forEach(v=>{
let b = text.includes(v)
if (b){
console.log(v)
}
})
res.forEach(v=>{
let b = text.includes(v)
if (b){
console.log(v)
}
})
// console.log(new Date().getTime()-startTime1+"ms");
</script>
</body>
</html>