com.mayabot.nlp.algorithm.collection.dat.DoubleArrayTrie Maven / Gradle / Ivy
/*
* Copyright 2018 mayabot.com authors. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
/**
* DoubleArrayTrieMap: Java implementation of Darts (Double-ARray Trie System)
*
*
* Copyright(C) 2001-2007 Taku Kudo <[email protected]>
* Copyright(C) 2009 MURAWAKI Yugo <[email protected]>
* Copyright(C) 2012 KOMIYA Atsushi <[email protected]>
*
*
*
* The contents of this file may be used under the terms of either of the GNU
* Lesser General Public License Version 2.1 or later (the "LGPL"), or the BSD
* License (the "BSD").
*
*/
/*
* 源代码参考和部分引用来自 https://github.com/hankcs/HanLP https://github.com/NLPchina/ansj_seg
*/
package com.mayabot.nlp.algorithm.collection.dat;
import java.io.DataInput;
import java.io.DataOutput;
import java.io.IOException;
import java.util.*;
import static com.mayabot.nlp.common.utils.DataInOutputUtils.readIntArray;
import static com.mayabot.nlp.common.utils.DataInOutputUtils.writeIntArray;
/**
* 双数组Trie树。
*
* 每个key对应其数值的下标（这个特性可以用来做TreeSet String()的索引）
*/
public class DoubleArrayTrie {
public int[] check;
public int[] base;
private int size;
public DoubleArrayTrie(DataInput in) throws IOException {
size = in.readInt();
base = readIntArray(in);
check = readIntArray(in);
}
public DoubleArrayTrie(TreeSet set) {
this(new ArrayList<>(set));
}
/**
* @param sortedKeys 有序且无重复
*/
public DoubleArrayTrie(List sortedKeys) {
DoubleArrayMaker datDoubleArrayMaker = new DoubleArrayMaker(sortedKeys);
datDoubleArrayMaker.build();
size = sortedKeys.size();
check = datDoubleArrayMaker.getCheck();
base = datDoubleArrayMaker.getBase();
}
public void write(DataOutput out) throws IOException {
out.writeInt(size);
writeIntArray(base, out);
writeIntArray(check, out);
}
/**
* 树叶子节点个数
*
* @return int
*/
public int size() {
return size;
}
/**
* @param text
* @param offset
* @return DATMatcher
*/
public DATMatcher matcher(String text, int offset) {
return new DATMatcher(this, text, offset);
}
/**
* @param text
* @param offset
* @return DATMatcher
*/
public DATMatcher matcher(char[] text, int offset) {
return new DATMatcher(this, text, offset);
}
/**
* @param text
* @return DATMatcher
*/
public DATMatcher matcher(String text) {
return new DATMatcher(this, text, 0);
}
/**
* @param text
* @return DATMatcher
*/
public DATMatcher matcher(char[] text) {
return new DATMatcher(this, text, 0);
}
/**
* @param text
* @param offset
* @return DATMatcher
*/
public DATLongMatcher matcherLong(String text, int offset) {
return new DATLongMatcher(this, text, offset);
}
/**
* @param text
* @param offset
* @return DATMatcher
*/
public DATLongMatcher matcherLong(char[] text, int offset) {
return new DATLongMatcher(this, text, offset);
}
/**
* @param text
* @return DATMatcher
*/
public DATLongMatcher matcherLong(String text) {
return new DATLongMatcher(this, text, 0);
}
/**
* @param text
* @return DATMatcher
*/
public DATLongMatcher matcherLong(char[] text) {
return new DATLongMatcher(this, text, 0);
}
/**
* 精确匹配
*
* @param key 键
* @return 值
*/
public int indexOf(CharSequence key) {
return indexOf(key, 0, 0, 0);
}
/**
* @param key
* @param pos
* @param len
* @param nodePos
* @return index
*/
public int indexOf(CharSequence key, int pos, int len, int nodePos) {
if (len <= 0) {
len = key.length();
}
if (nodePos <= 0) {
nodePos = 0;
}
int result = -1;
int b = base[nodePos];
int p;
for (int i = pos; i < pos + len; i++) {
p = b + (int) (key.charAt(i)) + 1;
if (b == check[p]) {
b = base[p];
} else {
return result;
}
}
p = b;
int n = base[p];
if (b == check[p] && n < 0) {
result = -n - 1;
}
return result;
}
/**
* 只查询一个char
*
* @param ch
* @return 位置 -1表示没有
*/
public int indexOf(char ch) {
int result = -1;
int b = base[0];
int p;
// for (int i = 0; i < 1; i++) {
p = b + (int) (ch) + 1;
if (b == check[p]) {
b = base[p];
} else {
return result;
}
// }
p = b;
int n = base[p];
if (b == check[p] && n < 0) {
result = -n - 1;
}
return result;
}
/**
* 精确查询
*
* @param chars 键的char数组
* @param pos char数组的起始位置
* @param len 键的长度
* 开始查找的位置（本参数允许从非根节点查询）
* @return 查到的节点代表的value ID，负数表示不存在
*/
public int indexOf(char[] chars, int pos, int len) {
return indexOf(chars, pos, len, 0);
}
/**
* 精确查询
*
* @param keyChars 键的char数组
* @param pos char数组的起始位置
* @param len 键的长度
* @param nodePos 开始查找的位置（本参数允许从非根节点查询）
* @return 查到的节点代表的value ID，负数表示不存在
*/
public int indexOf(char[] keyChars, int pos, int len, int nodePos) {
if (len <= 0) {
len = keyChars.length;
}
if (nodePos <= 0) {
nodePos = 0;
}
int result = -1;
int b = base[nodePos];
int p;
for (int i = pos; i < len + pos; i++) {
p = b + (int) (keyChars[i]) + 1;
if (b == check[p]) {
b = base[p];
} else {
return result;
}
}
p = b;
int n = base[p];
if (b == check[p] && n < 0) {
result = -n - 1;
}
return result;
}
public boolean containsKey(String key) {
return indexOf(key) >= 0;
}
/**
* 沿着路径转移状态
*
* @param path
* @return
*/
private int transition(String path) {
return transition(path.toCharArray());
}
/**
* 沿着节点转移状态
*
* @param path
* @return
*/
private int transition(char[] path) {
int b = base[0];
int p;
for (int i = 0; i < path.length; ++i) {
p = b + (int) (path[i]) + 1;
if (b == check[p]) {
b = base[p];
} else {
return -1;
}
}
p = b;
return p;
}
/**
* 转移状态
*
* @param c
* @param from
* @return
*/
private int transition(char c, int from) {
int b = from;
int p;
p = b + (int) (c) + 1;
if (b == check[p]) {
b = base[p];
} else {
return -1;
}
return b;
}
/**
* 沿着路径转移状态
*
* @param path 路径
* @param from 起点（根起点为base[0]=1）
* @return 转移后的状态（双数组下标）
*/
private int transition(String path, int from) {
int b = from;
int p;
for (int i = 0; i < path.length(); ++i) {
p = b + (int) (path.charAt(i)) + 1;
if (b == check[p]) {
b = base[p];
} else {
return -1;
}
}
p = b;
return p;
}
/**
* 检查状态是否对应输出
*
* @param state 双数组下标
* @return 对应的值，-1表示不输出
*/
public int output(int state) {
if (state < 0) {
return -1;
}
int n = base[state];
if (state == check[state] && n < 0) {
return -n - 1;
}
return -1;
}
/**
* 转移状态
*
* @param current
* @param c
* @return int
*/
protected int transition(int current, char c) {
int b = base[current];
int p;
p = b + c + 1;
if (b == check[p]) {
b = base[p];
} else {
return -1;
}
p = b;
return p;
}
public int getNonzeroSize() {
int result = 0;
for (int i = 0; i < check.length; ++i) {
if (check[i] != 0) {
++result;
}
}
return result;
}
public List commonPrefixSearch(String key) {
return commonPrefixSearch(key, 0, 0, 0);
}
/**
* 前缀查询
*
* @param key 查询字串
* @param pos 字串的开始位置
* @param len 字串长度
* @param nodePos base中的开始位置
* @return 一个含有所有下标的list
*/
public List commonPrefixSearch(String key, int pos, int len,
int nodePos) {
if (len <= 0) {
len = key.length();
}
if (nodePos <= 0) {
nodePos = 0;
}
List result = new ArrayList();
int b = base[nodePos];
int n;
int p;
for (int i = pos; i < len; i++) {
p = b + (int) (key.charAt(i)) + 1; // 状态转移 p = base[char[i-1]] +
// char[i] + 1
if (b == check[p]) // base[char[i-1]] == check[base[char[i-1]] +
// char[i] + 1]
{
b = base[p];
} else {
return result;
}
p = b;
n = base[p];
if (b == check[p] && n < 0) // base[p] == check[p] && base[p] < 0
// 查到一个词
{
result.add(-n - 1);
}
}
return result;
}
public List commonPrefixSearch(char[] key) {
return commonPrefixSearch(key, 0, 0, 0);
}
public List commonPrefixSearch(char[] key, int offset) {
return commonPrefixSearch(key, offset, 0, 0);
}
public List commonPrefixSearch(char[] key, int pos, int len,
int nodePos) {
if (len <= 0) {
len = key.length;
}
if (nodePos <= 0) {
nodePos = 0;
}
List result = new ArrayList();
int b = base[nodePos];
int n;
int p;
for (int i = pos; i < len; i++) {
p = b + (int) (key[i]) + 1; // 状态转移 p = base[char[i-1]] +
// char[i] + 1
if (b == check[p]) // base[char[i-1]] == check[base[char[i-1]] +
// char[i] + 1]
{
b = base[p];
} else {
return result;
}
p = b;
n = base[p];
if (b == check[p] && n < 0) // base[p] == check[p] && base[p] < 0
{ // 查到一个词
result.add(-n - 1);
}
}
return result;
}
/**
* 优化的前缀查询，可以复用字符数组
*
* @param keyChars
* @param begin
* @return LinkedList
*/
public LinkedList> commonPrefixSearchWithValue(char[] keyChars, int begin) {
int len = keyChars.length;
LinkedList> result = new LinkedList<>();
int b = base[0];
int n;
int p;
for (int i = begin; i < len; ++i) {
p = b;
n = base[p];
if (b == check[p] && n < 0) // base[p] == check[p] && base[p] < 0 查到一个词
{
result.add(new AbstractMap.SimpleEntry<>(new String(keyChars, begin, i - begin), -n - 1));
}
p = b + (int) (keyChars[i]) + 1; // 状态转移 p = base[char[i-1]] + char[i] + 1
// 下面这句可能产生下标越界，不如改为if (p < size && b == check[p])，或者多分配一些内存
if (b == check[p]) // base[char[i-1]] == check[base[char[i-1]] + char[i] + 1]
{
b = base[p];
} else {
return result;
}
}
p = b;
n = base[p];
if (b == check[p] && n < 0) {
result.add(new AbstractMap.SimpleEntry(new String(keyChars, begin, len - begin), -n - 1));
}
return result;
}
}