题目描述
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性: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
的节点。
提示:
0 <= index, val <= 1000
- 请不要使用内置的 LinkedList 库。
- 调用
get
、addAtHead
、addAtTail
、addAtIndex
和deleteAtIndex
的次数不超过2000
。
思路
直接实现对应的功能即可,不要忘记定义自己的链表节点。
链表操作的两种方式:
- 直接使用原来的链表来进行操作。
- 设置一个虚拟头结点在进行操作。
代码
C++版:
// 单链表
class MyLinkedList {
public:
// 定义链表节点结构体
struct LinkedNode {
int val;
LinkedNode* next;
LinkedNode(int val):val(val), next(nullptr){}
};
MyLinkedList() {
//定义虚拟头节点
myhead = new LinkedNode(0);
size=0;
}
int get(int index) {
//判断下标是否有效
if(index>(size-1) || index<0){
return -1;
}
LinkedNode* cur=myhead->next;
while(index--){
cur=cur->next;
}
return cur->val;
}
void addAtHead(int val) {
LinkedNode* newNode=new LinkedNode(val);
newNode->next=myhead->next;
myhead->next=newNode;
size++;
}
void addAtTail(int val) {
LinkedNode* newNode = new LinkedNode(val);
LinkedNode* cur=myhead;
while(cur->next!=nullptr){
cur=cur->next;
}
cur->next = newNode;
size++;
}
void addAtIndex(int index, int val) {
if(index>size) return;
if(index<0) index=0;
LinkedNode* newNode = new LinkedNode(val);
LinkedNode* cur=myhead;
while(index--) {
cur = cur->next;
}
newNode->next = cur->next;
cur->next = newNode;
size++;
}
void deleteAtIndex(int index) {
if(index>size-1 || index<0){
return;
}
LinkedNode* cur=myhead;
while(index--) {
cur = cur ->next;
}
LinkedNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
tmp=nullptr;
size--;
}
private:
int size;
LinkedNode* myhead;
};
Python版:
#单链表
class ListNode:
def __init__(self, val=0, next=None):
self.val = val
self.next = next
class MyLinkedList:
def __init__(self):
self.myhead = ListNode()
self.size = 0
def get(self, index: int) -> int:
if index < 0 or index >= self.size:
return -1
cur=self.myhead.next
for i in range(index):
cur=cur.next
return cur.val
def addAtHead(self, val: int) -> None:
self.myhead.next=ListNode(val,self.myhead.next)
self.size+=1
def addAtTail(self, val: int) -> None:
cur=self.myhead
while cur.next:
cur=cur.next
cur.next=ListNode(val)
self.size+=1
def addAtIndex(self, index: int, val: int) -> None:
if index < 0 or index > self.size:
return
cur=self.myhead
for i in range(index):
cur=cur.next
cur.next=ListNode(val,cur.next)
self.size += 1
def deleteAtIndex(self, index: int) -> None:
if index < 0 or index >= self.size:
return
cur=self.myhead
for i in range(index):
cur=cur.next
cur.next=cur.next.next
self.size-=1