順序棧和鏈棧的實現

這學期開了數據結構，剛開始以為代碼應該會很簡單，但是真正實現起來才發現有好多細節需要注意， 而且發現指針的姿勢忘了好多好多((٩(//̀Д/́/)۶))

 

130h.h

#include <bits/stdc++.h>
#define MAX 100
struct node1
{
    int *base;
    int *top;
    int Size;
};
struct node2
{
    int data;
    node2 *next;
};

void menu();
bool init1(node1 *&s);
void input1(node1 *s, int x);
bool pop1(node1 *s);
int top1(node1 *s);
bool init2(node2 *&s);
void input2(node2 *s, int x);
bool pop2(node2 *s);
int top2(node2 *s);

 

main.cpp

#include "130h.h"
using namespace std;

int main()
{
    int n, x, t, i;
    node1 *s1;
    node2 *s2;
    while(1)
    {
        menu();
        cin>>t;
        if(t==-1)
        {
            free(s1);
            free(s2);
            break;
        }
        switch(t)
        {
        case 1:
            if(init1(s1)) puts("yes");
            else puts("no");
            break;
        case 2:
            puts("輸入元素，-1結束");
            while(1)
            {
                scanf("%d", &x);
                if(x == -1) break;
                input1(s1, x);
            }
            puts("插入成功");
            break;
        case 3:
            puts("請輸入n的值");
            cin>>n;
            for(i = 1; i <= n; i++)
            {
                if(!pop1(s1))
                {
                    puts("棧空！");
                    break;
                }
            }
            if(i == n+1) puts("刪除成功");
            break;
        case 4:
            x = top1(s1);
            if(x == -1) puts("棧空！");
            else printf("%d\n", x);
            break;
        case 5:
            if(init2(s2)) puts("yes");
            else puts("no");
            break;
        case 6:
            puts("輸入元素，-1結束");
            while(1)
            {
                scanf("%d", &x);
                if(x == -1) break;
                input2(s2, x);
            }
            puts("插入成功");
            break;
        case 7:
            puts("請輸入n的值");
            cin>>n;
            for(i = 1; i <= n; i++)
            {
                if(!pop2(s2))
                {
                    puts("棧空");
                    break;
                }
            }
            if(i==n+1) puts("刪除成功");
            break;
        case 8:
            x = top2(s2);
            if(x == -1) puts("棧空！");
            else printf("%d\n", x);
            break;
        }
    }
    return 0;
}

 

function.cpp

#include "130h.h"

void menu()
{
    puts("");
    puts("-------順序棧操作-------");
    puts("1 初始化順序棧");
    puts("2 插入棧頂元素， -1結束");
    puts("3 刪除n個棧頂元素");
    puts("4 彈出棧頂元素");
    puts("-------鏈棧操作--------");
    puts("5 初始化鏈棧");
    puts("6 插入棧頂元素， -1結束");
    puts("7 刪除n個棧頂元素");
    puts("8 彈出棧頂元素");
    puts("------------------------");
    puts("-1 終止程序");
    puts("");
}

bool init1(node1 *&s)//初始化順序棧
{
    s = (node1*)malloc(sizeof(node1));
    s->base = (int*)malloc(MAX*sizeof(int));
    if(!s->base) return 0;
    s->top = s->base;
    s->Size = MAX;
    return 1;
}

void input1(node1 *s, int x)//插入棧頂元素
{
    if(s->top - s->base == MAX)
    {
        s->base = (int*)realloc(s->base, sizeof(int));
        s->top = s->base + MAX;
    }
    *s->top++ = x;
}

bool pop1(node1 *s)//刪除棧頂元素
{
    if(s->top == s->base) return 0;
    s->top--;
    return 1;
}

int top1(node1 *s)//彈出棧頂元素
{
    if(s->top == s->base) return -1;
    return *(s->top-1);
}

bool init2(node2 *&s)//初始化鏈棧
{
    s = (node2*)malloc(sizeof(node2));
    if(!s) return 0;
    s->next = NULL;
    return 1;
}

void input2(node2 *s, int x)////插入棧頂元素
{
    node2 *p;
    p = (node2*)malloc(sizeof(node2));
    p->data = x;
    p->next = s->next;
    s->next = p;
}

bool pop2(node2 *s)////刪除棧頂元素
{
    if(s->next == NULL) return 0;
    node2 *p = s->next;
    s->next = p->next;
    free(p);
    return 1;
}

int top2(node2 *s)////彈出棧頂元素
{
    if(!s->next) return -1;
    return s->next->data;
}

加油~