鏈棧

/**
	頭結點:如果沒有頭結點那麼插入和刪除就會多一種情況要處理。
	//這裏如果沒有頭結點,當root開始爲空,導致新建第一個結點找不到。
*/
#include<iostream>
using namespace std;

struct node {
	int data;
	node *link;
};

//聲明單鏈表頭結點
node *first;

/**
	單鏈表的查找:查找第k個結點(可包括頭結點,爲第0個)
*/

node *find(node *first,int k) {
	if(!first||k<0) {
		printf("find msg:error first==NULL||k<=0\n");
		return NULL;
	}
	node *temp=first;
	int i=0;
	while(temp&&i<k) {
		temp=temp->link;
		i++;
	}
	if(temp&&i==k) {
		printf("find msg:success\n");
		return temp;
	}
	printf("find msg:error find no\n");
	return NULL;
}

/**
	插入到第k個結點位置(k>=1)
*/

bool insert(node *first,int k,int data) {
	if(k<1) {
		printf("insert msg:error k<1\n");
		return false;
	}
	node *node_k=find(first,k-1);
	if(!node_k) {
		printf("insert msg:error node_k==NULL\n");
		return false;
	}
	node *newnode=new node();
	newnode->data=data;

	node *node_k_next=node_k->link;
	node_k->link=newnode;
	newnode->link=node_k_next;
	printf("insert msg:seccess\n");
	return true;
}

/**
	刪除第k個結點
*/

bool del(node *first,int k) {
	node *node_temp=find(first,k);
	if(!node_temp) {
		printf("del msg:error !node_temp\n");
		return false;
	}

	node *node_k_pre=find(first,k-1);
	if(!node_k_pre) {
		printf("del msg:error node_k==NULL\n");
		return false;
	}
	node *node_k=node_k_pre->link;
	if(!node_k->link) {
		delete node_k;
		//注意這裏釋放了node_k,node_k_pre=NULL,否則成野指針
		node_k_pre->link=NULL;
		printf("del msg:success\n");
		return true;
	}
	node *node_k_next=node_k->link;
	node_k_pre->link=node_k_next;
	delete node_k;
	printf("del msg:seccess\n");
	return true;
}

///////////////////////////////////////////////////////////////
//棧可以去掉頭結點了,這裏還是沒去掉
//top裏沒判斷空,默認有結點
int top() {
	return first->link->data;
}

bool empty() {
	if(!first->link)return true;
	return false;
}

void pop() {
	if(!empty()) {
		del(first,1);
	}
}

void push(int data) {
	insert(first,1,data);
}

int main() {
	//保證有頭結點
	first=new node();
	push(1);
	push(2);
	push(3);
	push(4);
	push(5);
	while(!empty()) {
		printf("data=%d\n",top());
		pop();
	}
	return 0;
}

發佈了81 篇原創文章 · 獲贊 8 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章