在链表中实现这些功能:
get(index):获取链表中第 index 个节点的值。如果索引无效,则返回-1。
addAtHead(val):在链表的第一个元素之前添加一个值为 val 的节点。插入后,新节点将成为链表的第一个节点。
addAtTail(val):将值为 val 的节点追加到链表的最后一个元素。
addAtIndex(index,val):在链表中的第 index 个节点之前添加值为 val 的节点。如果 index 等于链表的长度,则该节点将附加到链表的末尾。如果 index 大于链表长度,则不会插入节点。如果index小于0,则在头部插入节点。
deleteAtIndex(index):如果索引 index 有效,则删除链表中的第 index 个节点。
本文参考代码随想录
采用设置虚拟头结点的方式:
class MyLinkedList {
public:
struct LinkNode{
int val;
LinkNode* next;
LinkNode(int val) : val(val), next(NULL){}
};
MyLinkedList() {
_dummyHead = new LinkNode(0);
_size = 0;
}
int get(int index) {
if(index > (_size - 1) || index < 0){
return -1;
}
LinkNode* cur = _dummyHead -> next;
while(index--){
cur = cur->next;
}
return cur->val;
}
void addAtHead(int val) {
LinkNode* newNode = new LinkNode(val);
newNode->next = _dummyHead->next;
_dummyHead->next = newNode;
_size++;
}
void addAtTail(int val) {
LinkNode* newNode = new LinkNode(val);
LinkNode* cur = _dummyHead;
while(cur->next != NULL){
cur = cur->next;
}
cur->next = newNode;
_size++;
}
void addAtIndex(int index, int val) {
if(index > _size)return;
if(index < 0) index = 0;//在头部插入节点
LinkNode* newNode = new LinkNode(val);
LinkNode* cur = _dummyHead;
while(index--){
cur = cur->next;
}
newNode->next = cur->next;
cur->next = newNode;
_size++;
}
void deleteAtIndex(int index) {
if(index >= _size || index < 0){
return;
}
LinkNode* cur = _dummyHead;
while(index--){
cur = cur->next;
}
LinkNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
//被delete之后的指针tmp是随机值,必须加上tmp=NULL
tmp = NULL;
_size--;
}
void printLinkedList(){
LinkNode* cur = _dummyHead;
while(cur->next != NULL){
cout<<cur->next->val<<" ";
cur = cur->next;
}
cout<<endl;
}
private:
int _size;
LinkNode* _dummyHead;
};