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

题目: 给定一个单链表,一次反转k个节点,最终返回翻转后的链表的头节点;如果链表不足k个,则不变

举例:

Given this linked list: 1->2->3->4->5

For k = 2, you should return: 2->1->4->3->5

For k = 3, you should return: 3->2->1->4->5

解题思路:

1.  首先要确定链表的头节点newHead;即如果链表的节点个数大于k,则头节点为第k个节点;否则头节点不变;

2.  区分每一段,即确定段的开始和结尾;当前段的开始start, 段的结尾可以用k确定;下一段的开始为end;

3.  每k个节点之间的反转;

4.  段与段之间的连接,prevNode指向前一段的最后一个节点;

 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 }

 

posted @ 2016-09-17 22:20  小叶子leavescy  阅读(976)  评论(0编辑  收藏  举报