LeetCode接雨水和动物收容所

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();
 */