题目如下:
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性: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
提示:
0 <= index, val <= 1000
- 请不要使用内置的 LinkedList 库。
- 调用
get
、addAtHead
、addAtTail
、addAtIndex
和deleteAtIndex
的次数不超过2000
。
解题思路:可以使用直接法和虚拟节点法,这里我采用虚拟节点法方便进行操作,需要注意的尽量不要在插入函数里调用追加函数,防止记录长度的变量发生改变
代码如下:
class MyLinkedList {
public:
typedef ListNode Node;
typedef ListNode* pNode;
pNode virtualHead;
MyLinkedList() {//初始化
virtualHead = new ListNode{ 0 };
_size = 0;
}
int get(int index) {
if (index<0 || index>=_size )
{
return -1;
}
pNode temp = virtualHead->next;
while (index--)
{
temp = temp->next;
}
return temp->val;
}
void addAtHead(int val) {
//添加头结点
pNode head1 = new Node{val};
head1->next = virtualHead->next;
virtualHead->next = head1;
_size++;
}
void addAtTail(int val) {//追加到最后一个
pNode temp = virtualHead;
while (temp->next != NULL)
{
temp = temp->next;
}
pNode addNode = new Node{ val };
temp->next = addNode;
_size++;
}
void addAtIndex(int index, int val) {
//插入第index个节点前
pNode target = new Node{ val };
pNode temp = virtualHead;
if(index>_size)
{
return ;
}
if (index < 0) {
index = 0;
}
while (index--)
{
temp = temp->next;
}
target->next = temp->next;
temp->next = target;
_size++;
}
void deleteAtIndex(int index) {
if (index >= _size || index < 0)
{
return ;
}
pNode temp = virtualHead;
while (index--)
{
temp = temp->next;
}
pNode c = temp->next;
temp->next = temp->next-> next;
delete c;
c = NULL;
_size--;
}
private:
int _size;
};