Leetcode:203. 移除链表元素(C++)
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
示例 1:
输入:head = [1,2,6,3,4,5,6], val = 6 输出:[1,2,3,4,5]
示例 2:
输入:head = [], val = 1 输出:[]
示例 3:
输入:head = [7,7,7,7], val = 7 输出:[]
实现代码:
直接使用原来的链表来进行移除节点的操作
class Solution { public: ListNode* removeElements(ListNode* head, int val) { //删除头节点 while(head!=NULL&&head->val==val) { ListNode*temp=head; head=head->next; delete temp; } ListNode* p=head; //删除非头结点 while(p!=NULL&&p->next!=NULL) { if(p->next->val==val) { ListNode* temp=p->next; p->next=p->next->next; delete temp; } else { p=p->next; } } return head; } };
由于删除结点需要知道此结点的前驱结点,所以在删除过程中,删除头节点的代码就与其他结点不同,因为头结点没有前驱结点,要单独判断所要删除的结点是否为头节点,在删除头结点时要注意用的是while而不是if,因为在删除头结点后,新的头结点可能也为要删除的目标结点。同时要注意删除结点时指针变化的顺序,防止部分链表的丢失。注意这里头节点也储存元素。
设置一个虚拟头结点在进行移除节点的操作
class Solution { public: ListNode* removeElements(ListNode* head, int val) { ListNode* newhead=new ListNode();//建立一个新头节点 newhead->next=head; ListNode* p=newhead;//用于遍历的指针 while(p->next!=NULL) { if(p->next->val==val) { ListNode* temp=p->next; p->next=p->next->next; delete temp; } else { p=p->next; } } head=newhead->next; delete newhead; return head; } };
创建新的空头结点,可以让删除结点代码变得统一起来,返回的时候记得返回真正的头节点,而不是我们新创建的结点即可。
下一篇:
C语言编程—强制类型转换