算法学习笔记----前缀树、堆维持中位数、n皇后位运算加速

梦尘（0_O）

已于 2024-02-23 14:07:12 修改

阅读量1.7k

点赞数 53

文章标签： java 学习笔记算法数据结构

于 2024-02-22 16:17:00 首次发布

本文链接：https://blog.csdn.net/weixin_67661067/article/details/136236869

版权

前缀树

贪心（对数器、全排列模板）

堆维持中位数（随时添加数，动态取值随时可以获取中位数）

前缀树

前缀树：树节点有属性pass表示多少单词经过了当前字母，end表示有多少单词以当前字母结尾。属性nexts可以用数组表示，也可以用哈希表，若想有序可以用有序表。表示当前字母下的分叉字母前缀树添加：沿路所有字母节点pass++，如果没有节点就创建前缀树删除：沿路所有字母节点pass--，如果减为0就把指向这个节点的指针置空

package com.wtp.前缀树;

public class TrieNode {

    public int pass;
    public int end;
    public TrieNode[] nexts;
    
    //public HashMap<Char,Node> nexts;
    
    public TrieNode() {
        pass = 0;
        end = 0;
        nexts = new TrieNode[26];
    }
}

package com.wtp.前缀树;

public class Trie {

    private TrieNode root;
    
    public Trie() {
        root = new TrieNode();
    }
    
    public void insert(String word) {
        
        if(word == null) {
            return;
        }
        
        TrieNode cur = root;
        cur.pass++;
        char[] chs = word.toCharArray();
        for(int i = 0;i < chs.length;i++) {
            int index = chs[i] - 'a';
            if(cur.nexts[index] == null) {
                cur.nexts[index] = new TrieNode();
            }
            cur = cur.nexts[index];
            cur.pass++;
        }
        cur.end++;
    }
    
    //word这个单词之前加入过几次   
    public int search(String word) {
        if(word == null) {
            return 0;
        }
        
        TrieNode cur = root;
        char[] chs = word.toCharArray();
        for(int i = 0;i < chs.length;i++) {
            int index = chs[i] - 'a';
            if(cur.nexts[index] == null) {
                return 0;
            }
            cur = cur.nexts[index];
        }
        
        return cur.end;
    }
    
    //有多少个单词以pre作为前缀
    public int prefixNumber(String pre) {
        if(pre == null) {
            return 0;
        }
        
        TrieNode cur = root;
        char[] chs = pre.toCharArray();
        for(int i = 0;i < chs.length;i++) {
            int index = chs[i] - 'a';
            if(cur.nexts[index] == null) {
                return 0;
            }
            cur = cur.nexts[index];
        }
        return cur.pass;
    }
    
    public void del(String word) {
        if(search(word) != 0) {
            TrieNode cur = root;
            cur.pass--;
            char[] chs = word.toCharArray();
            for(int i = 0;i < chs.length;i++) {
                int index = chs[i] - 'a';
                if(--cur.nexts[index].pass == 0) {
                    cur.nexts[index] = null;
                    return;
                }
                cur = cur.nexts[index];
            }
            cur.end--;
        }
    }
}