432.全o(1)的数据结构

本文介绍了如何设计一个数据结构AllOne,它能存储字符串计数并支持增减操作以及查找最大最小计数的字符串。AllOne类使用双向链表来存储字符串,根据计数大小进行排序。inc()方法用于增加计数,dec()方法用于减少计数,当计数为0时删除字符串。getMaxKey()和getMinKey()分别返回计数最大和最小的字符串。示例展示了类的使用方式和预期输出。
摘要由CSDN通过智能技术生成

问题描述:
请你设计一个用于存储字符串计数的数据结构,并能够返回计数最小和最大的字符串。

实现 AllOne 类:

AllOne() 初始化数据结构的对象。
inc(String key) 字符串 key 的计数增加 1 。如果数据结构中尚不存在 key ,那么插入计数为 1 的 key 。
dec(String key) 字符串 key 的计数减少 1 。如果 key 的计数在减少后为 0 ,那么需要将这个 key 从数据结构中删除。测试用例保证:在减少计数前,key 存在于数据结构中。
getMaxKey() 返回任意一个计数最大的字符串。如果没有元素存在,返回一个空字符串 “” 。
getMinKey() 返回任意一个计数最小的字符串。如果没有元素存在,返回一个空字符串 “” 。

示例:

输入
[“AllOne”, “inc”, “inc”, “getMaxKey”, “getMinKey”, “inc”, “getMaxKey”, “getMinKey”]
[[], [“hello”], [“hello”], [], [], [“leet”], [], []]
输出
[null, null, null, “hello”, “hello”, null, “hello”, “leet”]

解释
AllOne allOne = new AllOne();
allOne.inc(“hello”);
allOne.inc(“hello”);
allOne.getMaxKey(); // 返回 “hello”
allOne.getMinKey(); // 返回 “hello”
allOne.inc(“leet”);
allOne.getMaxKey(); // 返回 “hello”
allOne.getMinKey(); // 返回 “leet”

提示:

1 <= key.length <= 10
key 由小写英文字母组成
测试用例保证:在每次调用 dec 时,数据结构中总存在 key
最多调用 inc、dec、getMaxKey 和 getMinKey 方法 5 * 104 次

public class AllOne {
    Node root;
    Dictionary<string, Node> nodes;

    public AllOne() {
        root = new Node();
        root.Prev = root;
        root.Next = root;  // 初始化链表哨兵,下面判断节点的 Next 若为 root,则表示 Next 为空(Prev 同理)
        nodes = new Dictionary<string, Node>();
    }
    
    public void Inc(string key) {
        if (nodes.ContainsKey(key)) {
            Node cur = nodes[key];
            Node nxt = cur.Next;
            if (nxt == root || nxt.Count > cur.Count + 1) {
                nodes[key] = cur.Insert(new Node(key, cur.Count + 1));
            } else {
                nxt.Keys.Add(key);
                nodes[key] = nxt;
            }
            cur.Keys.Remove(key);
            if (cur.Keys.Count == 0) {
                cur.Remove();
            }
        } else {  // key 不在链表中
            if (root.Next == root || root.Next.Count > 1) {
                nodes.Add(key, root.Insert(new Node(key, 1)));
            } else {
                root.Next.Keys.Add(key);
                nodes.Add(key, root.Next);
            }
        }
    }
    
    public void Dec(string key) {
        Node cur = nodes[key];
        if (cur.Count == 1) {  // key 仅出现一次,将其移出 nodes
            nodes.Remove(key);
        } else {
            Node pre = cur.Prev;
            if (pre == root || pre.Count < cur.Count - 1) {
                nodes[key] = cur.Prev.Insert(new Node(key, cur.Count - 1));
            } else {
                pre.Keys.Add(key);
                nodes[key] = pre;
            }
        }
        cur.Keys.Remove(key);
        if (cur.Keys.Count == 0) {
            cur.Remove();
        }
    }
    
    public string GetMaxKey() {
        if (root.Prev == null) {
            return "";
        }
        string maxKey = "";
        foreach (string key in root.Prev.Keys) {
            maxKey = key;
            break;
        }
        return maxKey;
    }
    
    public string GetMinKey() {
        if (root.Next == null) {
            return "";
        }
        string minKey = "";
        foreach (string key in root.Next.Keys) {
            minKey = key;
            break;
        }
        return minKey;
    }
}

class Node {
    public Node Prev { get; set; }
    public Node Next { get; set; }
    public ISet<string> Keys { get; set; }
    public int Count { get; set; }

    public Node() : this("", 0) {

    }

    public Node(string key, int count) {
        this.Count = count;
        Keys = new HashSet<string>();
        Keys.Add(key);
    }

    public Node Insert(Node node) {  // 在 this 后插入 node
        node.Prev = this;
        node.Next = this.Next;
        node.Prev.Next = node;
        node.Next.Prev = node;
        return node;
    }

    public void Remove() {
        this.Prev.Next = this.Next;
        this.Next.Prev = this.Prev;
    }
}
``
![在这里插入图片描述](https://img-blog.csdnimg.cn/d0f4ed603c6242669c6b88c94bb6aa39.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA5ZCh5ZCh5oCq,size_20,color_FFFFFF,t_70,g_se,x_16#pic_center)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值