关键词标记

关键词标记

    最近在做一个微信过滤的后台程序，设计的主要目的不是屏蔽你的消息，而是记录你的行为，别的就不能再说了！    平时的脏字过滤是把“你麻痹”替换成“你**”，而我的功能是标记出来，把“你麻痹”标记成“<#>你麻痹</#>”这样子，所以在做的时候遇到了不少的问题。一开始就想着用kmp算法实现嘛，不过做着做着心态就崩了，因为那边改需求，好吧！挺烦的。最后重新写了一遍，用了二叉树实现。实现的比较傻逼，别见怪。    原句：中国人爱中国的每一片土地。    关键词：{"中国人"，"爱"，"中国"}    我要返回的句子就是：        <#>中国人</#><#>爱</#><#>中国</#>的每一片土地。

//树节点：public class Node {    public String data;    public Node left;    public Node right;    public Node(String data) {        super();        this.data = data;          left=null;          right=null;      }}

这是一个很简单的树。
我的考虑是这样的：

我找到第一个词以后就对它进行分割，关键词就是父节点，然后子节点就是剩下的句子。 例如句子：    我们中国人爱中国人，爱中国的土地。关键词：    中国人，爱，中国（关键词按顺序标记出来）

思路，第一次找到中国人这个关键词，以它分割成三个元素：

分完以后，左边肯定没有中国人这个关键词，右边可能有，所以找右子树，找到后继续分：

继续找右子树看有没有中国人这个关键词，没有的话，这个关键词就找完了。
然后开始找第二个关键词：爱
这个时候要注意的是不能嵌套找，也就是说，你不能在关键词”中国人“里面标记“中国”这个关键词，所以，找的时候只能找叶子节点，如果有关键词，依旧像之前那样分割：

然后找中国

这样，这个树就搭建起来了：
之后通过前序遍历，遇到结点就添加标记符号<#>

// 先初始化树 根据第一个关键词把树先建立起来，左子树肯定没有第一个关键词，右子树有可能有关键词    public static void createTree(Node root, String word) {        // 当右子树不为空的时候判断，右子树为空的时候跳出循环        while (root.right != null) {            // 当右子树不含有关键词的时候函数返回            if (root.right.data.indexOf(word) != -1) {                root = root.right;                int indexOf = root.data.indexOf(word);                String l = root.data.substring(0, indexOf);                root.left = new Node(l);                String r = root.data.substring(indexOf + word.length(), root.data.length());                root.right = new Node(r);                root.data = word;            } else                break;        }    }    public static void rebuild(Node root, String word) { // 先根遍历        // 判断是否叶子节点        if (root != null) {            rebuild(root.left, word);            if (root.left == null && root.right == null) {                String msg = root.data;                int indexOf = msg.indexOf(word);                if (indexOf != -1) {                    String f = msg.substring(0, indexOf);                    String s = msg.substring(indexOf + word.length(), msg.length());                    root.left = new Node(f);                    root.right = new Node(s);                    root.data = word;                    createTree(root, word);                }            }            rebuild(root.right, word);        }    }    public static String preOrder(Node root,StringBuilder string) { // 先根遍历//      StringBuilder string = new StringBuilder();        if (root != null) {            preOrder(root.left,string);            if (root.left != null || root.right != null) {                string.append("<#>").append(root.data).append("</#>");//              System.out.print("<#>" + root.data + "</#>");            } else {//              System.out.print(root.data);                string.append(root.data);            }            // System.out.print(root.data+"-");            preOrder(root.right,string);        }        return string.toString();    }    public static void matchMsg(String msg, String[] word) {        Node root = new Node(word[0]);        int indexOf = msg.indexOf(word[0]);        if (indexOf != -1) {            String f = msg.substring(0, indexOf);            String s = msg.substring(indexOf + word[0].length(), msg.length());            root.left = new Node(f);            root.right = new Node(s);        }        createTree(root, word[0]);        for (int i = 1; i < word.length; i++) {            rebuild(root, word[i]);        }        System.out.println("先根遍历：");        StringBuilder string = new StringBuilder();        String pp = preOrder(root,string);        System.out.println(pp);    }    // 建立线程池并行处理    public static void poolThread(String msg,String[] word,String[] word1) {        ExecutorService fixedThreadPool = Executors.newFixedThreadPool(5);        for (int i = 1; i <= 2; i++) {            int index =i;            fixedThreadPool.execute(new Runnable() {                public void run() {                    if(index==1)matchMsg(msg,word);                    if(index==2)matchMsg(msg,word1);                }            });        }    }    public static void main(String[] args) {        String msg = "我们中国人爱中国人，爱中国的土地。；";        String[] word = { "中国人", "爱", "中国" };        String[] word1 = { "中国人"};        matchMsg(msg,word);    }