Add stack cpp

Author: xiangzhai

.gitignore

@@ -1,3 +1,5 @@
+.sconsign.dblite
+
 # Object files
 *.o
 

stack/SConstruct

@@ -1,3 +1,4 @@
 env = Environment(CCFLAGS='-g')
 
-env.Program('stack.c')
+env.Program('cstack.c')
+env.Program('stack.cpp')

stack/stack.c renamed to stack/cstack.c

@@ -116,9 +116,10 @@ static void m_travel()
 
     while (tmp) 
     {
-        printf("DEBUG: %d\n", tmp->data);
+        printf("%d\t", tmp->data);
         tmp = tmp->next;
     }
+    printf("\n");
 }
 
 static void m_cleanup() 
@@ -159,19 +160,19 @@ int main(int argc, char *argv[])
     {
         stack->push(i);
     }
-    printf("DEBUG: top %d ......\n", stack->top()->data);
-    printf("DEBUG: top %d ......\n", stack->top()->data);
-    printf("DEBUG: top %d ......\n", stack->top()->data);
-    printf("DEBUG: size %d ......\n", stack->size());
-    printf("DEBUG: travel ......\n");
+    printf("DEBUG: top %d\n", stack->top()->data);
+    printf("DEBUG: top %d\n", stack->top()->data);
+    printf("DEBUG: top %d\n", stack->top()->data);
+    printf("DEBUG: size %d\n", stack->size());
+    printf("DEBUG: travel\n");
     stack->travel();
-    printf("DEBUG: pop ......\n");
+    printf("DEBUG: pop\n");
     stack->pop();
-    printf("DEBUG: travel ......\n");
+    printf("DEBUG: travel\n");
     stack->travel();
-    printf("DEBUG: push 66 ......\n");
+    printf("DEBUG: push 66\n");
     stack->push(66);
-    printf("DEBUG: travel ......\n");
+    printf("DEBUG: travel\n");
     stack->travel();
     stack_cleanup(stack);
 

stack/stack.cpp

@@ -0,0 +1,133 @@
+#include <iostream>
+
+template<typename T> 
+class stack_element 
+{
+public:
+    stack_element() :next(NULL) {}
+    T data;
+    stack_element<T>* next;
+};
+
+template<typename T> 
+class stack 
+{
+public:
+    stack() :m_size(0), m_begin_element(NULL), m_current_element(NULL), 
+    m_top_element(NULL) {}
+
+    ~stack() 
+    {
+        stack_element<T>* tmp = m_begin_element;                                     
+                                                                                
+        while (tmp)                                                                 
+        {                                                                           
+            delete tmp;                                                              
+            tmp = tmp->next;                                                        
+        }                                                                           
+        
+        m_begin_element = NULL;                                                     
+        m_current_element = NULL;                                                   
+        m_top_element = NULL;
+    }
+
+    int size() 
+    {
+        return m_size;
+    }
+
+    stack_element<T>* top() 
+    {
+        if (m_top_element == NULL)                                                  
+            m_top_element = m_begin_element;                                        
+        else                                                                        
+            m_top_element = m_top_element->next;                                    
+                                                                                
+        return m_top_element;
+    }
+
+    void push(T data) 
+    {
+        stack_element<T>* tmp = new stack_element<T>();                     
+                                                                                
+        if (tmp == NULL)                                                            
+            return;                                                                 
+                                                                                
+        tmp->data = data;                                                           
+        tmp->next = NULL;                                                           
+                                                                                
+        if (m_begin_element == NULL)                                                
+            m_begin_element = tmp;                                                  
+        else                                                                        
+            m_current_element->next = tmp;                                          
+                                                                                
+        m_current_element = tmp;                                                    
+        m_size++;
+    }
+
+    void pop() 
+    {
+        stack_element<T>* tmp = NULL;                                                
+                                                                                
+        if (m_begin_element == NULL || m_current_element == NULL)                   
+            return;                                                                 
+                                                                                
+        tmp = m_begin_element;                                                      
+        while (tmp)                                                                 
+        {                                                                           
+            if (tmp->next == m_current_element)                                     
+            {                                                                       
+                delete tmp->next;                                                    
+                tmp->next = NULL;                                                   
+                m_current_element = tmp;                                            
+                m_size--;                                                          
+                break;                                                              
+            }                                                                       
+            tmp = tmp->next;                                                        
+        }
+    }
+
+    void travel() 
+    {
+        stack_element<T>* tmp = m_begin_element;                                     
+                                                                                
+        while (tmp)                                                                 
+        {                                                                           
+            std::cout << tmp->data << '\t';
+            tmp = tmp->next;                                                        
+        }
+        std::cout << std::endl;
+    }
+
+private:
+    int m_size;
+    stack_element<T>* m_begin_element;
+    stack_element<T>* m_current_element;
+    stack_element<T>* m_top_element;
+};
+
+int main(int argc, char* argv[]) 
+{
+    stack<int> stack_obj;
+    int i;
+    
+    for (i = 0; i < 10; i++) 
+    {
+        stack_obj.push(i);
+    }
+    std::cout << "DEBUG: top " << stack_obj.top()->data << std::endl;
+    std::cout << "DEBUG: top " << stack_obj.top()->data << std::endl;
+    std::cout << "DEBUG: top " << stack_obj.top()->data << std::endl;
+    std::cout << "DEBUG: travel" << std::endl;
+    stack_obj.travel();
+    std::cout << "DEBUG: pop" << std::endl;
+    stack_obj.pop();
+    std::cout << "DEBUG: travel" << std::endl;
+    stack_obj.travel();
+    std::cout << "DEBUG: push 66" << std::endl;
+    stack_obj.push(66);
+    std::cout << "DEBUG: travel" << std::endl;
+    stack_obj.travel();
+
+    return 0;
+}