40 lines
1008 B
C++
40 lines
1008 B
C++
#pragma once
|
||
|
||
|
||
class AcAuto
|
||
{
|
||
static const int kind = 256;
|
||
struct node{
|
||
node *fail; ///< 失败指针
|
||
union{
|
||
node *next[kind]; ///< Tire每个节点的256个子节点（以char存储，最多256字符）,
|
||
///< 构造fail指针的时候，作辅助作用时，next[0]保存树元素指针,next[1]保存下一个辅助节点
|
||
struct{
|
||
node* element;
|
||
node* nextAuxNode;
|
||
};
|
||
};
|
||
bool m_bUseAsAux; ///< 是否昨晚辅助节点
|
||
//node* pre, *post; ///< 构造fail指针的时候，作辅助作用
|
||
int count; ///< 是否为该单词的最后一个节点
|
||
node(bool bUseAsAux = false);
|
||
~node();
|
||
};
|
||
|
||
node* m_Root;
|
||
bool m_bBuild;
|
||
|
||
public:
|
||
AcAuto(void);
|
||
~AcAuto(void);
|
||
/// @param str 输入的字符串
|
||
/// @param bOnlyCheckHasKeyWorld 只是简单检查是否有关键字，不需要统计共有多少个
|
||
int Query(const char* str, bool bOnlyCheckHasKeyWorld = false);
|
||
|
||
void InsertKeyWord(const char* keyWord);
|
||
|
||
/// 设置 fail 指针
|
||
void BuildAcAutomation();
|
||
};
|
||
|