例题
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性: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
的节点。
虚拟头结点
// 时间复杂度: 涉及 index 的相关操作为 O(index), 其余为 O(1)
// 空间复杂度: O(n)
class MyLinkedList {
public:
// 此题需要自己定义节点结构
struct LinkedNode
{
int val;
LinkedNode *next;
LinkedNode(int val):val(val), next(NULL){}
};
// 定义链表构造函数
MyLinkedList() {
dummyHead=new LinkedNode(0);
size=0;
}
int get(int index) {
if(index<0 || index>=size)
return -1;
LinkedNode *cur=dummyHead->next;
for(int i=0;i<index;i++)
{
cur=cur->next;
}
return cur->val;
}
void addAtHead(int val) {
LinkedNode *newNode=new LinkedNode(val);
// 注意顺序
newNode->next=dummyHead->next;
dummyHead->next=newNode;
size++;
}
void addAtTail(int val) {
LinkedNode *newNode=new LinkedNode(val);
// 注意这里cur指针必须指向虚拟头指针,因为此时链表可能为空
// 直接指向dummyHead->next会使得cur变成一个野指针
LinkedNode *cur=dummyHead;
while(cur->next!=NULL)
cur=cur->next;
cur->next=newNode;
size++;
}
void addAtIndex(int index, int val) {
if(index>size)
return;
LinkedNode *newNode=new LinkedNode(val);
LinkedNode *cur=dummyHead;
for(int i=0;i<index;i++)
cur=cur->next;
newNode->next=cur->next;
cur->next=newNode;
size++;
}
void deleteAtIndex(int index) {
if(index>=size || index<0)
return;
LinkedNode *cur=dummyHead;
for(int i=0;i<index;i++)
cur=cur->next;
LinkedNode *tmp=cur->next;
cur->next=cur->next->next;
delete tmp;
size--;
}
private:
int size;
LinkedNode *dummyHead;
};
/**
* 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);
*/