use template class for Stack_LL.cpp

Author: ashwek

3_Data_Structures/10_Elementary_Data_Structures/Stack_LL.cpp

@@ -1,88 +1,117 @@
 #include <iostream>
+#include <stdlib.h>
 
 using namespace std;
 
+template<class DataType>
 class Node{
-public:
-int value;
-Node *next;
-
-Node(int val, Node* ptr){
-value = val;
-next = ptr;
-}
-
-~Node(){
-free(next);
-}
+public:
+ DataType value;
+ Node<DataType> *next;
+
+ Node(DataType val, Node<DataType>* ptr) {
+ value = val;
+ next = ptr;
+ }
+
+ ~Node(){
+ free(next);
+ }
+
 };
 
+template<class DataType>
 class Stack_LL{
+private:
+ Node<DataType> *top;
+ int len;
+
+public:
+
+ Stack_LL(){
+ top = NULL;
+ len = 0;
+ }
+
+ ~Stack_LL(){
+ Node<DataType>* tempNode;
+ while( top != NULL ){
+ tempNode = top;
+ top = top->next;
+ tempNode->next = NULL;
+ free(tempNode);
+ }
+ }
+
+ bool is_empty(){
+ return (top == NULL);
+ }
+
+ void push(DataType value) {
+ top = new Node<DataType>(value, top);
+ len++;
+ }
+
+ DataType pop(){
+
+ if( is_empty() )
+ return -1;
+
+ DataType tempValue = top->value;
+
+ Node<DataType> *tempNode = top;
+ top = top->next;
+ tempNode->next = NULL;
+ free(tempNode);
+ len--;
+
+ return tempValue;
+
+ }
+
+ int length(){
+ return len;
+ }
+
+ void display(){
+ if( ! is_empty() ){
+ Node<DataType> *temp = top;
+ while( temp != NULL){
+ cout<<temp->value <<", ";
+ temp = temp->next;
+ }
+ }
+ }
 
-private:
-Node *top;
-int len;
-
-public:
-Stack_LL(){
-top = NULL;
-len = 0;
-}
-
-~Stack_LL(){
-free(top);
-}
-
-int is_empty(){
-return (top == NULL);
-}
-
-void push(int value){
-top = new Node(value, top);
-len++;
-}
-
-int pop(){
-if( is_empty() )
-return -1;
-Node *temp = top;
-top = top->next;
-len--;
-return temp->value;
-}
-
-int length(){
-return len;
-}
-
-void display(){
-if( ! is_empty() ){
-Node *temp = top;
-while( temp != NULL){
-cout<<temp->value <<", ";
-temp = temp->next;
-}
-}
-}
 };
 
+
 int main(){
 
-Stack_LL S1;
+ srand(time(NULL));
+
+ Stack_LL<int> S1;
+ int temp;
+ cout <<"Push 5 random values in stack :";
+ for(int i = 0; i < 5; i++) {
+ temp = random() % 100;
+ cout<<"\nPushing : " <<temp;
+ S1.push(temp);
+ }
 
-for(int i=1; i<5; i++){
-cout<<"\nPush " <<(i*10);
-S1.push(i*10);
-}
+ cout<<"\n\nStack = ";
+ S1.display();
 
-cout<<"\n\nStack = ";
-S1.display();
+ cout<<"\n\nPop: popped value = " <<S1.pop();
 
-cout<<"\n\nPop, popped value = " <<S1.pop();
+ cout<<"\n\nStack = ";
+ S1.display();
 
-cout<<"\n\nStack = ";
-S1.display();
+ cout <<"\n\nPop stack until empty :";
+ while( ! S1.is_empty() ){
+ cout <<"\nPopped : " <<S1.pop();
+ }
 
-cout<<endl;
-return 0;
+ cout<<endl;
+ return 0;
 }