add linked list stack

2021-04-15 23:43:44 +08:00 · 2021-04-15 23:43:44 +08:00 · 6c8b22b204
commit 6c8b22b204
parent fa4748c57c
1 changed files with 184 additions and 0 deletions
--- a/src/main/java/io/github/ehlxr/datastructure/stack/LinkedListStackDemo.java
+++ b/src/main/java/io/github/ehlxr/datastructure/stack/LinkedListStackDemo.java
@ -0,0 +1,184 @@
+/*
+ * The MIT License (MIT)
+ *
+ * Copyright © 2021 xrv <xrg@live.com>
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+
+package io.github.ehlxr.datastructure.stack;
+
+import java.util.Scanner;
+
+/**
+ * @author ehlxr
+ * @since 2021-04-15 22:18.
+ */
+public class LinkedListStackDemo {
+
+    public static void main(String[] args) {
+        //测试一下ArrayStack 是否正确
+        //先创建一个ArrayStack对象->表示栈
+        LinkedListStack stack = new LinkedListStack(4);
+        String key;
+        boolean loop = true; //控制是否退出菜单
+        Scanner scanner = new Scanner(System.in);
+
+        while (loop) {
+            System.out.println("show: 表示显示栈");
+            System.out.println("exit: 退出程序");
+            System.out.println("push: 表示添加数据到栈(入栈)");
+            System.out.println("pop: 表示从栈取出数据(出栈)");
+            System.out.println("请输入你的选择");
+            key = scanner.next();
+            switch (key) {
+                case "show":
+                    stack.list();
+                    break;
+                case "push":
+                    System.out.println("请输入一个数");
+                    int value = scanner.nextInt();
+                    stack.push(value);
+                    break;
+                case "pop":
+                    try {
+                        int res = stack.pop();
+                        System.out.printf("出栈的数据是 %d\n", res);
+                    } catch (Exception e) {
+                        // TODO: handle exception
+                        System.out.println(e.getMessage());
+                    }
+                    break;
+                case "exit":
+                    scanner.close();
+                    loop = false;
+                    break;
+                default:
+                    break;
+            }
+        }
+
+        System.out.println("程序退出~~~");
+    }
+
+}
+
+class LinkedListStack {
+    private final int maxSize;
+    private int top = -1;
+    private final Node head = new Node(0);
+
+    public LinkedListStack(int maxSize) {
+        this.maxSize = maxSize;
+    }
+
+    public boolean isFull() {
+        return maxSize == top + 1;
+    }
+
+    public boolean isEmpty() {
+        return top == -1;
+    }
+
+    public void push(int value) {
+        if (isFull()) {
+            System.out.println("栈满");
+            return;
+        }
+        Node temp = head;
+        while (true) {
+            if (temp.next == null) {
+                temp.next = new Node(value);
+                top++;
+                break;
+            }
+            temp = temp.next;
+        }
+    }
+
+    public int pop() {
+        if (isEmpty()) {
+            throw new RuntimeException("栈空");
+        }
+
+        Node temp = head;
+        while (true) {
+            if (temp.next == null) {
+                throw new RuntimeException("栈空");
+            }
+            if (temp.next.next == null) {
+                top--;
+                int id = temp.next.id;
+                temp.next = null;
+                return id;
+            }
+            temp = temp.next;
+        }
+
+    }
+
+    public void list() {
+        if (isEmpty()) {
+            System.out.println("栈空");
+            return;
+        }
+
+        revert();
+
+        Node temp = head.next;
+        while (temp != null) {
+            System.out.println(temp);
+
+            temp = temp.next;
+        }
+
+        revert();
+
+    }
+
+    private void revert() {
+        Node pre = null;
+        Node cur = head.next;
+        while (cur != null) {
+            Node temp = cur;
+            cur = cur.next;
+
+            temp.next = pre;
+            pre = temp;
+        }
+
+        head.next = pre;
+    }
+
+}
+
+
+class Node {
+    public int id;
+    public Node next;
+
+    public Node(int id) {
+        this.id = id;
+    }
+
+    @Override
+    public String toString() {
+        return "Node{" + id + "}";
+    }
+}