LeetCode--025--k个一组翻转链表(java and python)

给出一个链表，每 k 个节点一组进行翻转，并返回翻转后的链表。

k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么将最后剩余节点保持原有顺序。

示例 :

给定这个链表：1->2->3->4->5

当 k = 2 时，应当返回: 2->1->4->3->5

当 k = 3 时，应当返回: 3->2->1->4->5

说明 :

• 你的算法只能使用常数的额外空间。
• 你不能只是单纯的改变节点内部的值，而是需要实际的进行节点交换。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
        if(head == null || head.next == null) return head;
        int count = 0;
        ListNode cur = head;
        while(cur != null && count != k){
            cur = cur.next;
            count++;
        }
        if(count == k){
            cur = reverseKGroup(cur,k);//cur 是上一次反转后的头结点
            while(count-- >0){//基本的反转语句要掌握
                ListNode temp = head.next;
                head.next = cur;
                cur = head;
                head = temp;
            }
            head = cur;
        }
        return head;
    }
}

2019-04-20 17:50:52

python版本

start到end为当前要翻转的部分

class Solution:
    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        dummy=pre=start=end = ListNode(-1)
        dummy.next=head
        while end.next!=None:
            for i in range(k):
                if end!=None:
                    end=end.next
                else:
                    break
            if end==None:
                break
            start = pre.next
            next=end.next
            end.next=None
            pre.next = self.reverse(start)
            start.next=next
            pre=start
            end=pre
        return dummy.next
    def reverse(self,cur):
        pre = ListNode(-1)
        pre.next=None
        while cur!=None:
            q = cur.next
            cur.next=pre
            pre=cur
            cur = q
        return pre