题解 | #牛群的重新分组#
牛群的重新分组
https://www.nowcoder.com/practice/267c0deb9a6a41e4bdeb1b2addc64c93
import java.util.*;
/*
* public class ListNode {
* int val;
* ListNode next = null;
* public ListNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param head ListNode类
* @param k int整型
* @return ListNode类
*/
public ListNode reverseKGroup (ListNode head, int k) {
// write code here
if (head == null || k == 1) {
return head; // 链表为空或k为1,不需要进行翻转操作
}
ListNode dummy = new ListNode(-1); // 增加一个虚拟头节点,简化操作
dummy.next = head;
ListNode prev = dummy;
ListNode curr = head;
int count = 0;
ListNode temp = head;
while (temp != null) {
count++;
temp = temp.next;
}
while (count >= k) {
ListNode tail = prev.next; // 每个分组的第一个节点,也是上一轮翻转后的末尾节点
for (int i = 1; i < k; i++) {
ListNode nextGroup = curr.next;
curr.next = nextGroup.next;
nextGroup.next = prev.next;
prev.next = nextGroup;
}
prev = tail; // 更新prev为翻转后的分组末尾
curr = tail.next; // 更新curr为下一个待翻转的分组的第一个节点
count -= k;
}
return dummy.next; // 返回去掉虚拟头节点后的链表
}
}
题目考察的知识点:
链表操作、链表分组翻转、节点交换。
解题方法分析:
- 首先判断特殊情况,如果链表为空或者k为1,直接返回原链表,因为不需要进行翻转操作。
- 创建一个虚拟头节点
dummy,将其指向链表的头节点head,用它来方便处理头节点的情况。 - 定义两个指针
prev和curr,初始时都指向dummy节点。prev用来记录每个分组的前一个节点,curr用来遍历分组中的节点。 - 使用一个循环,遍历链表。在每一轮循环中,需要判断剩余的节点数量是否大于等于k个,如果不够k个,就不进行翻转操作。
- 如果剩余节点数量大于等于k个,开始进行分组翻转操作。使用一个辅助指针
tail来标记每个分组的最后一个节点。 - 在分组翻转过程中,记录
curr节点后面的节点为nextGroup,然后断开curr与其后面的节点的连接。 - 对这k个节点进行翻转操作,将这k个节点按照逆序连接到
prev节点之后,然后将tail节点连接到翻转后的分组末尾。 - 更新
prev为tail,将curr指向nextGroup,继续下一轮分组翻转操作。 - 循环结束后,返回处理后的链表,即从
dummy.next开始。
这段代码通过循环遍历链表,每k个节点进行一次翻转操作,最后返回处理后的链表。在每一轮翻转中,使用prev和curr指针来记录分组的前一个节点和当前节点,然后使用tail标记每个分组的末尾节点。通过断开和连接节点的操作来实现分组翻转。