707.设计链表
在链表类中实现这些功能:
- get(index):获取链表中第 index 个节点的值。如果索引无效,则返回-1。
- addAtHead(val):在链表的第一个元素之前添加一个值为 val 的节点。插入后,新节点将成为链表的第一个节点。
- addAtTail(val):将值为 val 的节点追加到链表的最后一个元素。
- addAtIndex(index,val):在链表中的第 index 个节点之前添加值为 val 的节点。如果 index 等于链表的长度,则该节点将附加到链表的末尾。如果 index 大于链表长度,则不会插入节点。如果index小于0,则在头部插入节点。
- deleteAtIndex(index):如果索引 index 有效,则删除链表中的第 index 个节点。
public class ListNode{
int val;
ListNode next;
ListNode(int val){this.val = val;}
}
class MyLinkedList {
int size = 0;
ListNode head;
public MyLinkedList() {
head = new ListNode(0);
}
public int get(int index) {
if(index < 0|| index >= size) return -1; // 下标从0开始
ListNode curr = head.next;
while(index-->0) curr = curr.next;
return curr.val;
}
public void addAtHead(int val) {
addAtIndex(0, val);
}
public void addAtTail(int val) {
addAtIndex(size, val);
}
public void addAtIndex(int index, int val) {
if(index > size) return;
if(index < 0) index = 0;
size++;
ListNode pre = head;
while(index-->0) pre = pre.next;
ListNode curr = new ListNode(val);
curr.next = pre.next;
pre.next = curr;
}
public void deleteAtIndex(int index) {
if(index<0||index>=size) return;
size--;
ListNode pre = head;
while(index-->0) pre = pre.next;
pre.next = pre.next.next;
}
}