Acwing826单链表

题目

实现一个单链表，链表初始为空，支持三种操作：

1. 向链表头插入一个数；
2. 删除第 k 个插入的数后面的数；
3. 在第 k 个插入的数后插入一个数。

现在要对该链表进行 M 次操作，进行完所有操作后，从头到尾输出整个链表。

注意:题目中第 k 个插入的数并不是指当前链表的第 k 个数。例如操作过程中一共插入了 n 个数，则按照插入的时间顺序，这 n 个数依次为：第 11 个插入的数，第 22 个插入的数，...第 n 个插入的数。

输入格式

第一行包含整数 M，表示操作次数。

接下来 M 行，每行包含一个操作命令，操作命令可能为以下几种：

1. H x，表示向链表头插入一个数 x
2. D k，表示删除第 k 个插入的数后面的数（当 k 为 00 时，表示删除头结点）。
3. I k x，表示在第 k 个插入的数后面插入一个数 x（此操作中 k 均大于 0）。

输出格式

共一行，将整个链表从头到尾输出。

数据范围

1≤M≤100000 所有操作保证合法。

代码与思路

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    static int idx, head, N = 100010;
    static int[] e = new int[N], ne = new int[N];

    /**
     * 初始化数组链表
     */
    static void init() {
        head = -1;
        idx = 0;
    }
    /**
     * 插入到头结点
     * @param x
     */
    static void add_to_head(int x) {
        e[idx] = x;
        ne[idx] = head;
        head = idx ++;
    }
    /**
     * 插入到第k个结点
     * @param k
     * @param x
     */
    static void add(int k, int x) {
        e[idx] = x;
        ne[idx] = ne[k];
        ne[k] = idx ++;
    }
    /**
     * 移除第k个结点后面的链表
     * @param k
     */
    static void remove(int k) {
         ne[k] = ne[ne[k]];
    }
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        init();
        while (n > 0) {
            n--;
            String s = in.next();
            char ts = s.charAt(0);
            if(ts == 'H') {
                int x = in.nextInt();
                add_to_head(x);
            } else if (ts == 'D') {
                int k = in.nextInt();
                if(k == 0)  head = ne[head];  //这个分支一定要加，要不然会越界
                else remove(k - 1);
            } else {
                int k = in.nextInt();
                int x = in.nextInt();
                add(k - 1, x);
            }
        }
        for (int ix = head; ix != -1; ix = ne[ix] ){
            System.out.print(e[ix] + " ");
        }
    }

}