请你实现一个数据结构支持以下操作:
Inc(key) - 插入一个新的值为 1 的 key。或者使一个存在的 key 增加一,保证 key 不为空字符串。
Dec(key) - 如果这个 key 的值是 1,那么把他从数据结构中移除掉。否则使一个存在的 key 值减一。如果这个 key 不存在,这个函数不做任何事情。key 保证不为空字符串。
GetMaxKey() - 返回 key 中值最大的任意一个。如果没有元素存在,返回一个空字符串"" 。
GetMinKey() - 返回 key 中值最小的任意一个。如果没有元素存在,返回一个空字符串""。
挑战:
你能够以 O(1) 的时间复杂度实现所有操作吗?
解题思路
1,这是lru的变体,用hash+有序双链表的形式
2,每次inc和dec的时候将当前节点从链表中去除
3,然后找到合适的位置插入
4,注意,当inc后,后面的连续常数个值相等,需要特殊处理
5,dec类似
代码实现
type Node struct{
next *Node
prev *Node
val int
key string
}
type AllOne struct {
head *Node
tail *Node
m map[string]*Node
}
/** Initialize your data structure here. */
func Constructor() AllOne {
ao:=AllOne{
head:&Node{},
tail:&Node{},
m:make(map[string]*Node),
}
ao.head.next=ao.tail
ao.tail.prev=ao.head
return ao
}
/** Inserts a new key <Key> with value 1. Or increments an existing key by 1. */
func (this *AllOne) Inc(key string) {
if n,ok:=this.m[key];ok{
this.m[key].val++
next:=n.next
if next==this.tail || next.val>=n.val{
return
}else{
next.prev=n.prev
n.prev.next=next
for next.next!=this.tail && next.next.val<n.val{
next=next.next
}
n.next=next.next
n.prev=next
next.next.prev=n
next.next=n
}
}else{
n:=&Node{
val:1,
key:key,
prev:this.head,
next:this.head.next,
}
this.head.next.prev=n
this.head.next=n
this.m[key]=n
}
this.Print()
}
func(this*AllOne)Print(){
for n:=this.head.next ;n!=this.tail;n=n.next{
println(n.val)
}
}
/** Decrements an existing key by 1. If Key's value is 1, remove it from the data structure. */
func (this *AllOne) Dec(key string) {
if _,ok:=this.m[key];!ok {
return
}
n:= this.m[key]
if this.m[key].val>1{
n.val--
if n.prev==this.head || n.prev.val<=n.val{
return
}else {
prev:=n.prev
prev.next=n.next
n.next.prev=prev
for prev.prev!=this.head && prev.prev.val>n.val{
prev=prev.prev
}
n.prev=prev.prev
n.next=prev
prev.prev.next=n
prev.prev=n
}
}else{
n.prev.next=n.next
n.next.prev=n.prev
delete(this.m,key)
}
this.Print()
}
/** Returns one of the keys with maximal value. */
func (this *AllOne) GetMaxKey() string {
if this.head.next==this.tail{
return ""
}
return this.tail.prev.key
}
/** Returns one of the keys with Minimal value. */
func (this *AllOne) GetMinKey() string {
if this.head.next==this.tail{
return ""
}
return this.head.next.key
}
/**
* Your AllOne object will be instantiated and called as such:
* obj := Constructor();
* obj.Inc(key);
* obj.Dec(key);
* param_3 := obj.GetMaxKey();
* param_4 := obj.GetMinKey();
*/
本文分享自 golang算法架构leetcode技术php 微信公众号,前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!