設計數據結構SetOfStacks， 由多個棧組成，而且在前一個棧填滿時新建一個棧

設想有一堆盤子，堆過高可能會倒下來。所以，現實生活中，盤子堆到必定高度時，咱們就會另外堆一堆盤子。請事先設計數據結構SetOfStacks， 模擬這種行爲。

SetOfStacks應該由多個棧組成，而且在前一個棧填滿時新建一個棧。此外，SetOfStacks.push() 和SetOfStacks.pop()應該與普通棧的操做方法相同（也就是說，

pop（）返回的值，應該跟只有一個棧時的狀況同樣）。

 

1. import java.util.Stack;  
2.   
3.   
4. public class SetOfStacks {  
5.     class stack {  
6.         public void push(int v);  
7.         public int pop();  
8.     }  
9.       
10.     //push and pop are both operated from the last stack, just pay attention if the last stack is empty or full  
11.     public void push(int value) {  
12.         Stack<E> last = getLastStack();  
13.         if(last != null && !last.isFull()) {  
14.             last.push(value);  
15.         }  
16.         else {  
17.             Stack stack = new Stack(capacity);  
18.             stack.push(value);  
19.             stacks.add(stack);  
20.         }  
21.     }  
22.       
23.     public int pop() {  
24.         Stack last = getLastStack();  
25.         int v = last.pop();  
26.         if (last.size == 0 ) {  
27.             stacks.remove(stacks.size() -1);  
28.         }  
29.         return -1;  
30.     }  
31. }  

 

進階：實現一個popAt( int index)方法，根據指定的子棧，執行pop操做。

1. public class SetOfStacks{  
2.   
3.       ArrayList<MyStack2> stacks= new ArrayList<MyStack2>();  
4.       int capacity,numOfStack;  
5.       public static void main(String[] args) {  
6.              // TODO Auto-generated method stub  
7.   
8.       }  
9.   
10.       public void push(int val){  
11.             MyStack2 last=getLastStack();  
12.              if( last!= null &&! last.isFull())  
13.                    last.push( val);  
14.              else{  
15.                   MyStack2 stack= new MyStack2( capacity);  
16.                    stack.push( val);  
17.                    stacks.add( stack);  
18.                    numOfStack++;  
19.             }  
20.       }  
21.         
22.       public int pop(){  
23.             MyStack2 last=getLastStack();  
24.              int value= last.pop();  
25.              if( last. size==0)  
26.                    stacks.remove( last); //移除  
27.              return value;  
28.       }  
29.   
30.       /** 
31.        * 思路：從棧1彈出元素時，移出棧2的棧底元素壓入棧1中，隨後將棧3的棧底元素壓入棧2的棧頂…… 
32.        *          （假定出最後一個棧外，其他棧都是滿的） 
33.        * @param index 
34.        * @return 
35.        */  
36.       public int popAt(int index){  
37.              return leftShift( index, true);  
38.       }  
39.         
40.       public int leftShift( int index, boolean removeTop){  
41.             MyStack2 stack= stacks.get( index);  
42.              int removedItem;  
43.              if( removeTop)  
44.                    removedItem= stack.pop();  
45.              else  
46.                    removedItem= stack.removeBottem();  
47.               
48.              if( stack.isEmpty()){  
49.                    stacks.remove( stack);  
50.             } else if(! stacks.get( index+1).isEmpty()){  
51.                    int v=leftShift( index+1, false);  
52.                    stack.push( v);  
53.             }  
54.              return removedItem;  
55.       }  
56.         
57.       public MyStack2 getLastStack() {  
58.              // TODO Auto-generated method stub  
59.              return stacks.get( numOfStack-1);  
60.       }  
61. }  
62.   
63. class MyStack2{  
64.       private int capacity;  
65.       public Node top,bottem;  
66.       public int size;  
67.         
68.       public MyStack2( int capacity){  
69.              this. capacity= capacity;  
70.       }  
71.         
72.       public boolean isFull(){  
73.              return capacity== size;  
74.       }  
75.         
76.       public void join(Node above,Node below){  
77.              if( below!= null)  
78.                    below. above= above;  
79.              if( above!= null)  
80.                    above. below= below;  
81.       }  
82.         
83.       public boolean push(int value){  
84.              if( size>= capacity)  
85.                    return false;  
86.              size++;  
87.             Node n= new Node( value);  
88.              if( size==1)  
89.                    this. bottem= n;  
90.             join( n, this. top);  
91.              top= n;  
92.               
93.              return true;  
94.       }  
95.         
96.       public int pop(){  
97.             Node n= top;  
98.              top= top. below;  
99.              size--;  
100.              return n. data;  
101.       }  
102.         
103.       public boolean isEmpty(){  
104.              return size==0;  
105.       }  
106.         
107.       public int removeBottem(){  
108.             Node b= bottem;  
109.              bottem= bottem. above;  
110.              if( bottem!= null)  
111.                    bottem. below= null;  
112.              size--;  
113.              return b. data;  
114.       }  
115. }