題目: 實現(xiàn)一個棧,帶有出棧(pop),入棧(push),取最小元素(getMin)三個方法。要保證這三個方法的時間復(fù)雜度都是O(1)。
設(shè)原有的棧叫做棧A,此時創(chuàng)建一個額外的棧B,用于輔助原棧A。
當(dāng)?shù)谝粋€元素進(jìn)入棧A的時候,讓新元素的下標(biāo)進(jìn)入棧B。把這個第一個元素當(dāng)做是棧A的當(dāng)前最小值。(考慮到棧中元素可能不是類對象,所以B棧存儲的是A棧元素的下標(biāo))
每當(dāng)新元素進(jìn)入棧A時,比較新元素和棧A當(dāng)前最小值的大小,如果小于棧A當(dāng)前最小值,則讓新元素的下標(biāo)進(jìn)入棧B,此時棧B的棧頂元素就是棧A當(dāng)前最小值的下標(biāo)。
每當(dāng)棧A有元素出棧時,如果出棧元素是棧A當(dāng)前最小值,則讓棧B的棧頂元素也出棧。此時棧B余下的棧頂元素所指向的,是棧A當(dāng)中原本第二小的元素,代替剛才的出棧元素成為了棧A的當(dāng)前最小值。(備胎轉(zhuǎn)正)
當(dāng)調(diào)用getMin方法的時候,直接返回棧B的棧頂所指向的棧A對應(yīng)元素即可。

這個解法中進(jìn)棧、出棧、取最小值的時間復(fù)雜度都是O(1),最壞情況空間復(fù)雜度是O(N)。