剑指 Offer 09. 用两个栈实现队列(相邻相同操作只需要移动一次)
生活随笔
收集整理的這篇文章主要介紹了
剑指 Offer 09. 用两个栈实现队列(相邻相同操作只需要移动一次)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目
用兩個棧實現一個隊列。隊列的聲明如下,請實現它的兩個函數 appendTail 和 deleteHead ,分別完成在隊列尾部插入整數和在隊列頭部刪除整數的功能。(若隊列中沒有元素,deleteHead 操作返回 -1 )
示例 1:
輸入: ["CQueue","appendTail","deleteHead","deleteHead"] [[],[3],[],[]] 輸出:[null,null,3,-1] 示例 2:輸入: ["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"] [[],[],[5],[2],[],[]] 輸出:[null,-1,null,null,5,2]提示:
1 <= values <= 10000
最多會對 appendTail、deleteHead 進行 10000 次調用
思路
建立兩個棧,一個主棧,一個輔助棧,主棧用于插入,輔助棧用于刪除
插入時,如果輔助棧為空,說明上一次操作為插入,直接在主棧中插入即可,反之需要將輔組棧中數據從尾到頭依次搬到主棧中
刪除時,如果主棧為空,說明上一次操作為刪除,直接在輔助棧中刪除,反之搬數據
優點
相鄰操作相同,只需要搬動一次數據即可
代碼
type CQueue struct {stack []intsubStack []int }func Constructor() CQueue {return CQueue{stack : make([]int, 0),subStack : make([]int, 0),} }func (this *CQueue) AppendTail(value int) {// 插入時,如果輔助棧為空,直接在主棧進行插入if len(this.subStack) == 0 {this.stack = append(this.stack, value)return}// 將輔助棧中數據從尾到頭搬到主棧中,進行插入endIndex := len(this.subStack) -1for ; endIndex >= 0; endIndex--{this.stack = append(this.stack, this.subStack[endIndex])}this.stack = append(this.stack, value)// 清空輔助棧this.subStack = this.subStack[:0]}func (this *CQueue) DeleteHead() int {// 主棧與輔助棧都為空,隊列中沒有數據,返回-1if len(this.stack) == 0 && len(this.subStack) == 0{return -1}value := int(0)// 主棧為空,直接在輔助棧中進行刪除if len(this.stack) == 0 {endIndex := len(this.subStack)-1value = this.subStack[endIndex]this.subStack = this.subStack[:endIndex]return value}// 主棧有數據,從尾到頭將數據搬到輔助棧后,進行刪除endIndex := len(this.stack) -1for ; endIndex > 0; endIndex--{this.subStack = append(this.subStack, this.stack[endIndex])}value = this.stack[0]this.stack = this.stack[:0]return value}/*** Your CQueue object will be instantiated and called as such:* obj := Constructor();* obj.AppendTail(value);* param_2 := obj.DeleteHead();*/總結
以上是生活随笔為你收集整理的剑指 Offer 09. 用两个栈实现队列(相邻相同操作只需要移动一次)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 分段线性变换与直方图修正
- 下一篇: Pycharm下安装Tensorflow