题目描述:
设计链表的实现。您可以选择使用单链表或双链表。单链表中的节点应该具有两个属性:val 和 next。val 是当前节点的值,next 是指向下一个节点的指针/引用。如果要使用双向链表,则还需要一个属性 prev 以指示链表中的上一个节点。假设链表中的所有节点都是 0-index 的。
在链表类中实现这些功能:
get(index):获取链表中第 index 个节点的值。如果索引无效,则返回-1。
addAtHead(val):在链表的第一个元素之前添加一个值为 val 的节点。插入后,新节点将成为链表的第一个节点。
addAtTail(val):将值为 val 的节点追加到链表的最后一个元素。
addAtIndex(index,val):在链表中的第 index 个节点之前添加值为 val 的节点。如果 index 等于链表的长度,则该节点将附加到链表的末尾。如果 index 大于链表长度,则不会插入节点。如果index小于0,则在头部插入节点。
deleteAtIndex(index):如果索引 index 有效,则删除链表中的第 index 个节点。
#include <iostream>
using namespace std;
class MyLinkedList {
public:
//定义链表结点的结构体
struct List
{
int val;
List *next;
};
//成员变量
List *myHead;//自设的虚拟头结点
int size;//链表长度
//构造函数初始化成员变量
MyLinkedList() {
myHead = new List;//新建一个虚拟头结点,并初始化
myHead->val = 0; myHead->next = NULL;
size = 0;//链表长度初始化
}
int get(int index) {
/*我的写法
int n = 0;
List *pMark = myHead->next;
while (pMark != NULL)
{
if (n == index)
{
return pMark->val;
}
else
{
n++;
pMark = pMark->next;
}
}
return -1;
*/
//《代码随想录》的写法
if (index < 0 || index > size -1)
return -1;
//以下循环代码的通用功能:找到并指向下标为index的结点
List *pMark = myHead->next;
while (index--)
{
pMark = pMark->next;
}
//此时pMark指向下标为index的结点,返回其内部的值
return pMark->val;
}
void addAtHead(int val) {
//先new一个结点
List *pAdd = new List;
pAdd->val = val;
pAdd->next = NULL;
//头插法
pAdd->next = myHead->next;
myHead->next = pAdd;
size++;
}
void addAtTail(int val) {
//先new一个结点
List *pAdd = new List;
pAdd->val = val;
pAdd->next = NULL;
//尾插法
List *pMark = myHead;
while (pMark->next != NULL)
{
pMark = pMark->next;
}
pMark->next = pAdd;
size++;
}
void addAtIndex(int index, int val) {
//先new一个结点
List *pAdd = new List;
pAdd->val = val;
pAdd->next = NULL;
//如果index > size ,直接返回
if (index > size)
return;
//以下循环代码的通用功能:找下标为index的前一个位置
//(特殊情况也成立:例如原来有三个结点,size=3,下标范围0-2,当index=3时,插入位置为尾结点的后面,即下标为3,循环结束后pPre会停留在尾结点处,即下标为2处)
List *pPre = myHead;//pPre为待插入位置的前一个
while (index--)
{
pPre = pPre->next;
}
//将待插入结点pAdd插入pPre之后
pAdd->next = pPre->next;
pPre->next = pAdd;
size++;
}
void deleteAtIndex(int index) {
/*我的写法
List *pDel = NULL;
List *pPre = myHead;
int n = 0;
while (pPre->next != NULL)
{
if (n == index)
{
pDel = pPre->next;
pPre->next = pDel->next;
delete pDel;
size--;
return;
}
else
{
n++;
pPre = pPre->next;
}
}
*/
//《代码随想录》的写法,用以上提到的循环
if (index < 0 || index > size -1)
return;
List *pDel = NULL;
//用以上提到的循环
List *pPre = myHead;
while (index--)
{
pPre = pPre->next;
}
pDel = pPre->next;
pPre->next = pDel->next;
delete pDel;
size--;
}
};
int main()
{
MyLinkedList linkedList; //创建链表对象linkedList
int res = linkedList.get(0); //初始链表为空,故res为-1
linkedList.addAtHead(1); //添加元素1
linkedList.addAtTail(3); //添加元素3,此时链表为[1,3]
linkedList.addAtIndex(1,2); //插入元素2,此时链表为[1,2,3]
int res2 = linkedList.get(1); //下标为1,故返回2
int res3 = linkedList.get(4); //index无效,返回-1
linkedList.deleteAtIndex(2); //删除下标为2的元素,故此时链表为[1,2]
system("pause");
return 0;
}