首页 > 代码库 > Reverse Nodes in k-Group

Reverse Nodes in k-Group

2024-07-02 15:50:16   223人阅读

Link: http://oj.leetcode.com/problems/reverse-nodes-in-k-group/

Given a linked list, reverse the nodes of a linked list k at a time and return its modified list.

If the number of nodes is not a multiple of k then left-out nodes in the end should remain as it is.

You may not alter the values in the nodes, only nodes itself may be changed.

Only constant memory is allowed.

For example,
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 /**
 2  * Definition for singly-linked list.
 3  * public class ListNode {
 4  *     int val;
 5  *     ListNode next;
 6  *     ListNode(int x) {
 7  *         val = x;
 8  *         next = null;
 9  *     }
10  * }
11  */
12 public class Solution {
13     public ListNode reverseKGroup(ListNode head, int k) {
14         if (head == null || head.next == null)
15             return head;
16         ListNode pre = new ListNode(0);
17         pre.next = head;
18         head = pre;
19         int index = 1;
20         ListNode cur = pre.next;
21         ListNode post = cur.next;
22         //before execution, we need to check if there‘s enough element
23         while (enoughElement(cur, k)) {
24             while (index < k) {
25                 ListNode temp = post.next;
26                 post.next = pre.next;
27                 cur.next = temp;
28                 pre.next = post;
29                 post = temp;
30                 index++;
31             }
32             //after the reverse operation,we need to initial
33             //the parameter
34             index = 1;
35             pre = cur;
36             cur = cur.next;
37             //note the cur may be null
38             if (cur != null) {
39 
40                 post = cur.next;
41             }
42         }
43         return head.next;
44 
45     }
46 
47     public boolean enoughElement(ListNode head, int k) {
48         int count = 0;
49         //note the head could be null, then we cannot use head.next
50         while (head != null) {
51             head = head.next;
52             count++;
53         }
54         if (count < k)
55             return false;
56         return true;
57 
58     }
59 }

 

For the basic idead of the algorithm, please refer to http://www.cnblogs.com/Altaszzz/p/3704780.html

 

 


联系
我们