Leetcode25--->Reverse Nodes in k-Group(以k个节点为段，反转单链表)

 1 /**
 2  * Definition for singly-linked list.
 3  * public class ListNode {
 4  *     int val;
 5  *     ListNode next;
 6  *     ListNode(int x) { val = x; }
 7  * }
 8  */
 9 public class Solution {
10     public ListNode reverseKGroup(ListNode head, int k) {
11         if(head == null || k <= 1)
12             return head;
13         ListNode begin = head; // begin是当前段的开始
14         ListNode end = begin;  // end是下一段的开始
15         ListNode newHead = begin;  // 新的链表头
16         ListNode prevNode = head;  // 是上一段的最后一个节点
17         int flag = 0;  // 记录是第几段
18         while(end != null){
19             int count = 1;
20             while(end.next != null && count < k){
21                 end = end.next;
22                 count ++;
23             }
24             if(count == k){
25                 if(flag == 0){
26                     newHead = end;
27                 }
28                 flag ++;
29                 if(flag > 1){  // 不是第一段时就需要段与段之间的连接
30                     prevNode.next = end;  // 段和段之间进行连接
31                     prevNode = begin;
32                 }
33                 end = end.next;  // 下一个段的开始
34                 ListNode prev = null;
35                 int i = 0;
36                 while(i < k){  // 链表的反转
37                     ListNode nextNode = begin.next;
38                     begin.next = prev;
39                     prev = begin;
40                     begin = nextNode;
41                     i++;
42                 }
43                 begin = end;   // 下一个段的开始
44                 prevNode.next = end;  // 两个段之间连接，是为了避免下一段节点数不足k个，则不能使用第30行代码来连接段
45             }
46             else
47                 break;
48         }
49         return newHead;
50     }
51 }