牛客网高频算法题系列-BM3-链表中的节点每k个一组翻转
题目描述
将给出的链表中的节点每 k 个一组翻转,返回翻转后的链表如果链表中的节点数不是 k 的倍数,将最后剩下的节点保持原样你不能更改节点中的值,只能更改节点本身。
原题目见:BM3 链表中的节点每k个一组翻转
解法一:递归
使用递归的方式处理每k个一组的结点,具体处理方式如下:
- 首先用tail记录每次翻转的尾部,就是第k+1个结点,从头结点开始,遍历链表,找到第k+1个结点,如果当前结点数不够k个,则不用反转直接返回;
- 然后从链表头开始遍历,翻转结点并记录反转后的表头pre和表尾结点head;
- 然后递归处理从tail开始的后面的链表结点,并将head指向递归处理后的链表;
- 返回新的表头即为反转后的新的链表的表头结点。
代码
public class Bm003 {
/**
* 递归
*
* @param head ListNode类
* @param k int整型
* @return ListNode类
*/
public static ListNode reverseKGroup(ListNode head, int k) {
// 找到每次翻转的尾部
ListNode tail = head;
// 遍历k次到尾部
for (int i = 0; i < k; i++) {
// 如果不足k到了链表尾部,直接返回,不翻转
if (tail == null) {
return head;
}
tail = tail.next;
}
// 翻转时需要的前序和当前节点
ListNode pre = null, cur = head;
// 在到达当前段尾部节点前
while (cur != tail) {
ListNode temp = cur.next;
cur.next = pre;
pre = cur;
cur = temp;
}
// 当前尾部指向下一段要翻转的链表
head.next = reverseKGroup(tail, k);
return pre;
}
public static void main(String[] args) {
ListNode head = ListNode.testCase2();
System.out.println("翻转之前");
ListNode.print(head);
// 当k=2时
ListNode newHead = reverseKGroup(head, 2);
System.out.println("当k=2时,翻转之后");
ListNode.print(newHead);
ListNode head1 = ListNode.testCase2();
// 当k=3时
ListNode newHead2 = reverseKGroup(head1, 3);
System.out.println("当k=3时,翻转之后");
ListNode.print(newHead2);
}
}
相信坚持的力量!
网友评论