golang刷leetcode 技巧（26）堆盘子

type StackOfPlates struct {
  cap int
  data [][]int
}


func Constructor(cap int) StackOfPlates {
    return StackOfPlates{cap:cap}
}


func (this *StackOfPlates) Push(val int)  {
    if this.cap==0{
        return
    }
    i:=len(this.data)
    if i==0 || len(this.data[i-1])==this.cap{
       this.data=append(this.data,[]int{val})
    }else{
       this.data[i-1]=append(this.data[i-1],val)
    }
}


func (this *StackOfPlates) Pop() int {
   l:= len(this.data)
   if l==0{
       return -1
   }
   l1:=len(this.data[l-1])
   val:=this.data[l-1][l1-1]
   if l1==1{
       this.data=this.data[:l-1:l-1]
   }else{
       this.data[l-1]=this.data[l-1][:l1-1:l1-1]
   }
   return val
}


func (this *StackOfPlates) PopAt(index int) int {
   l:=len(this.data)
   if index>l-1{
       return -1
   } 
   l1:=len(this.data[index])
   val:=this.data[index][l1-1]
   if l1==1{
       this.data=append(this.data[:index:index],this.data[index+1:]...)
   }else{
       this.data[index]=this.data[index][:l1-1:l1-1]
   }
   return val
}


/**
 * Your StackOfPlates object will be instantiated and called as such:
 * obj := Constructor(cap);
 * obj.Push(val);
 * param_2 := obj.Pop();
 * param_3 := obj.PopAt(index);
 */