题目描述:
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性:val 和 next 。val 是当前节点的值,next 是指向下一个节点的指针/引用。
如果是双向链表,则还需要属性 prev 以指示链表中的上一个节点。假设链表中的所有节点下标从 0 开始。
实现 MyLinkedList 类:
- MyLinkedList() 初始化 MyLinkedList 对象。
- int get(int index) 获取链表中下标为 index 的节点的值。如果下标无效,则返回 -1 。
- void addAtHead(int val) 将一个值为 val 的节点插入到链表中第一个元素之前。在插入完成后,新节点会成为链表的第一个节点。
- void addAtTail(int val) 将一个值为 val 的节点追加到链表中作为链表的最后一个元素。
- void addAtIndex(int index, int val) 将一个值为 val 的节点插入到链表中下标为 index 的节点之前。如果 index 等于链表的长度,那么该节点会被追加到链表的末尾。如果 index 比长度更大,该节点将 不会插入 到链表中。
- void deleteAtIndex(int index) 如果下标有效,则删除链表中下标为 index 的节点。
示例:
输入
[“MyLinkedList”, “addAtHead”, “addAtTail”, “addAtIndex”, “get”, “deleteAtIndex”, “get”]
[[], [1], [3], [1, 2], [1], [1], [1]]
输出
[null, null, null, null, 2, null, 3]
解释
MyLinkedList myLinkedList = new MyLinkedList();
myLinkedList.addAtHead(1);
myLinkedList.addAtTail(3);
myLinkedList.addAtIndex(1, 2); // 链表变为 1->2->3
myLinkedList.get(1); // 返回 2
myLinkedList.deleteAtIndex(1); // 现在,链表变为 1->3
myLinkedList.get(1); // 返回 3
代码实现:
class MyLinkedList {
int size;// 链表长度
ListNode head;// 头节点
public MyLinkedList() {
// 初始化链表:长度为0,头节点为空
size = 0;
head = null;
}
public int get(int index) {
if (index < 0 || index > size - 1) {// 索引越界,直接返回
return -1;
}
ListNode dummy = new ListNode();
dummy.next = head;// 定义虚拟头节点,指向头节点
ListNode cur = dummy.next;// 用于遍历的节点
while (index > 0) {// 找到索引下标对应的节点
cur = cur.next;
index--;
}
return cur.val;
}
public void addAtHead(int val) {
addAtIndex(0, val);// 头插法:也就是在索引为0的节点前面插入
}
public void addAtTail(int val) {
ListNode dummy = new ListNode();// 虚拟节点:防止链表初始为空的情况,导致空指针
dummy.next = head;// 指向头节点
ListNode cur = dummy;// 遍历节点
while (cur.next != null) {
cur = cur.next;// 找到尾插的插入位置
}
cur.next = new ListNode(val);// 插入新节点
head = dummy.next;// 更新头指针
size++;
}
public void addAtIndex(int index, int val) {
ListNode dummy = new ListNode();
dummy.next = head;// 定义虚拟头节点,指向头节点
ListNode cur = dummy;// 用于遍历的节点
ListNode newNode = new ListNode(val);// 定义新节点
if (index <= size) {// index取值合法才允许插入
while (index > 0) {
cur = cur.next;// 虚拟节点开始遍历的当前节点的后插 等同于 索引节点的前插
index--;
}
newNode.next = cur.next;// 新节点指向当前节点的后继
cur.next = newNode;// 当前节点再指向 新节点
size++;
}
head = dummy.next;// 更新头指针
}
public void deleteAtIndex(int index) {
if (index >= size) {// 索引越界的情况
return;
}
ListNode dummy = new ListNode();
dummy.next = head;// 定义虚拟头节点,指向头节点
ListNode cur = dummy;// 用于遍历的节点
while (index > 0) {
cur = cur.next;// 找到删除的位置
index--;
}
if (cur.next != null) {// 如果后继不为空,代表可以删除
cur.next = cur.next.next;// 当前指针指向后继的后继
size--;
head = dummy.next;// 更新头节点,因为存在删除0索引的情况
}
}
}
/**
* Your MyLinkedList object will be instantiated and called as such:
* MyLinkedList obj = new MyLinkedList();
* int param_1 = obj.get(index);
* obj.addAtHead(val);
* obj.addAtTail(val);
* obj.addAtIndex(index,val);
* obj.deleteAtIndex(index);
*/
// 定义节点类:数值及其后继
public class ListNode {
int val;
ListNode next;
ListNode() {
}
ListNode(int val) {
this.val = val;
}
ListNode(int val, ListNode next) {
this.val = val;
this.next = next;
}
}