Go语言实现算法,打卡
接雨水
题目
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
Code
// 接雨水
func trap(height []int) int {
n := len(height)
if 0 == n {
return 0
}
leftMax := make([]int, n)
leftMax[0] = height[0]
for i:=1;i<n;i++ {
leftMax[i] = max(leftMax[i-1],height[i])
}
rightMax := make([]int,n)
rightMax[n-1] = height[n-1]
for i:=n-2;i>=0;i-- {
rightMax[i] = max(rightMax[i+1],height[i])
}
ret:=0
for i:=0;i<n;i++ {
ret += min(leftMax[i],rightMax[i]) - height[i]
}
return ret
}
func max(a, b int) int {
if a > b {
return a
}
return b
}
func min(a, b int) int {
if a < b {
return a
}
return b
}
动物收容所
题目
动物收容所。有家动物收容所只收容狗与猫,且严格遵守“先进先出”的原则。在收养该收容所的动物时,收养人只能收养所有动物中“最老”(由其进入收容所的时间长短而定)的动物,或者可以挑选猫或狗(同时必须收养此类动物中“最老”的)。换言之,收养人不能自由挑选想收养的对象。请创建适用于这个系统的数据结构,实现各种操作方法,比如enqueue、dequeueAny、dequeueDog和dequeueCat。允许使用Java内置的LinkedList数据结构。
enqueue方法有一个animal参数,animal[0]代表动物编号,animal[1]代表动物种类,其中 0 代表猫,1 代表狗。
dequeue*方法返回一个列表[动物编号, 动物种类],若没有可以收养的动物,则返回[-1,-1]。
Code
// 动物收容所
type AnimalShelf struct {
dogList *list.List
catList *list.List
timeList *list.List
}
func Constructor() AnimalShelf {
return AnimalShelf{
list.New(),
list.New(),
list.New(),
}
}
func (this *AnimalShelf) Enqueue(animal []int) {
var l *list.List
if animal[1] == 0 {
l=this.catList
} else {
l=this.dogList
}
ele := this.timeList.PushBack(animal)
l.PushBack(ele)
}
func (this *AnimalShelf) DequeueAny() []int {
if this.timeList.Len() == 0 {
return []int{-1, -1}
}
animal := this.timeList.Remove(this.timeList.Front()).([]int)
var l *list.List
if animal[1] == 0 {
l = this.catList
} else {
l = this.dogList
}
l.Remove(l.Front()) // element.Value should equals timeline.Front()
return animal
}
func (this *AnimalShelf) DequeueDog() []int {
if this.dogList.Len() == 0 {
return []int{-1, -1}
}
ele := this.dogList.Remove(this.dogList.Front()).(*list.Element)
return this.timeList.Remove(ele).([]int)
}
func (this *AnimalShelf) DequeueCat() []int {
if this.catList.Len() == 0 {
return []int{-1, -1}
}
ele := this.catList.Remove(this.catList.Front()).(*list.Element)
return this.timeList.Remove(ele).([]int)
}
/**
* Your AnimalShelf object will be instantiated and called as such:
* obj := Constructor();
* obj.Enqueue(animal);
* param_2 := obj.DequeueAny();
* param_3 := obj.DequeueDog();
* param_4 := obj.DequeueCat();
*/