本文已参与「新人创作礼」活动,一起开启掘金创作之路。
题
定义一个函数,输入一个链表的头节点,反转该链表并输出反转后链表的头节点。
输入: 1->2->3->4->5->NULL
输出: 5->4->3->2->1->NULL
/**
* Definition for singly-linked list.
* function ListNode(val) {
* this.val = val;
* this.next = null;
* }
*/
/**
* @param {ListNode} head
* @return {ListNode}
*/
var reverseList = function(head) {
};
解
首先考链表,我们要对链表解构以及栈堆内存十分清晰。
我们要知道我们对链表进行遍历是必然的,所以可以写出
var reverseList = function(head) {
let curr = head;
while (curr) {
//进入下一个循环
curr=curr.next
}
};
当我们当头部开始时,头部成了尾部,应该指向null,所以我们应该创建一个null节点,所以第一次遍历时,是头部指向一个null。
但是当我们修改了当前节点的next之后,链表被切断,我们就不能用它进行遍历了,所以我们得要保存被切断之后的链表,声明一个next指针保存
var reverseList = function(head) {
let res = null
let curr = head;
while (curr) {
const next=curr.next
curr.next=res
//进入下一个循环我们要思考一个问题...
curr=curr.next
}
};
上述代码说了进入下一个循环前我们要思考一个问题。
当我们遍历到第二步时,我们希望的是curr能够指向1那个节点,可是没有栈内存也就是没有指针指向它,因此我们要在遍历到下一步时,可以用res指向它。
var reverseList = function(head) {
let res = null
let curr = head
while (curr) {
const next=curr.next
curr.next=res
//进入下一个循环前该做的事,对应图中步骤1,然后开始循环
res=curr
curr=next
}
};
res就会指向反转链表的头部,返回即可
var reverseList = function(head) {
let res = null
let curr = head
while (curr) {
const next=curr.next
curr.next=res
res=curr
curr=next
}
return res
};
提升
实现每k个一组反转?
- 将反转定义一个方法。
- 在头部声明一个
dummy指向head,作为第一组需要反转的链表的头部,并额外声明两个指针pre和end。- 用
end.next作为循环结束的条件。- 将
end移动k个链,若在移动中变为空说明已经没有可以旋转的k个链了,跳出循环。- 将
end.next置为空,意思就是让原链表从end处截断,但是在反转链表中我们已经学过了,我们应该声明一个next把end后面的链表保存,不然后面内容就丢失了。- 用
start指向pre.next,用于反转,因为end在k个链处被截断了,所以刚好反转k个。然后用pre接收,也就是把反转的内容连在dummy后面,反转完,start指向的位置是dummy链的尾部,我们只需要让start.next指向原来保存的next,也就是start作为新一组要反转的链表的头部,将pre和end指向它进入循环。
var reverseKGroup = function(head, k) {
//解释1
const rev=(start)=>{
let res=null
while(start!==null){
const next=start.next
start.next=res
res=start
start=next
}
return res
}
//解释2
const dummy=new ListNode(0,head)
let pre=dummy,end=dummy
//解释3
while(end.next!==null){
//解释4
for(let i=0;i<k&&end!==null;i++) end=end.next
if(end===null) break
//解释5
const next=end.next
end.next=null
//解释6
let start=pre.next
pre.next=rev(start)
start.next=next
pre=start
end=start
}
return dummy.next
};