最近好忙啊,先是人工智能要交一个研究报告,然后我的JAVA暑期实训又来了,JAVA实训是做一个养老系统,听说要完成起码3000多行代码。今天抽空写一道 leetcode吧
给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。
列表中的节点数目在范围 [0, 104] 内
很不喜欢用指针,所以我写链表题之后可能是用JAVA。
先放一个JAVA链表基本的构造:
public class ListCode {
int val;
ListCode next;
public ListCode(){};
public ListCode(int val){
this.val = val;
}
public ListCode(int val,ListCode next){
this.val = val;
this.next = next;
}
}
这种数组一般是双链表,但是在构造里不用同时定义上一个节点和下一个,
我们可以单独定义一个会变化的 pre;
class Solution {
public ListNode removeElements(ListNode head, int val) {
if(head == null)//如果为空就返回空
return head;
ListNode d = new ListNode(-1,head);//定义一个真正的头节点:没有val值(此时赋值-1是因为-1不在数据范围内)
ListNode pre = d;//定义一个指向前一个节点的
ListNode cur = head;//从第一个元素开始检查
while(cur != null){
if(cur.val == val)
pre.next = cur.next;//基本的删除操作
else{
pre = cur; //如果没删上一个节点就变了,删了的话就没变
}
cur = cur.next; //移动判断下一个
}
return d.next;
}
}
补充:这几天又写了一次,但是出现了一些问题,下面我将分析一下问题出在哪里:
class Solution {
public ListNode removeElements(ListNode head, int val) {
if(head == null)
return head;
ListNode dom = new ListNode();
ListNode cur = new ListNode();
cur = dom;
while(head != null){
if(head.val == val){
head = head.next;
}
else{
dom.next = head;
dom = dom.next;
head = head.next;
}
}
return cur.next;
}
}
主体思想是用 dom 去接收符合要求的值,但是我们发现下面这个样例出现了问题:
那就是当前一个元素不是 val,后面的元素都是 val 时,我们就会出现问题。这其实是JAVA的引用问题
dom.next = head;//以此题为例,最后的节点指向 val = 5的节点
dom = dom.next;
head = head.next;//head 由指向 val = 5的节点变为指向 val = 6,
//但是dom 的指向没有变,之后不可能走 else 程序,故最后会出错