【Leetcode 25】K 个一组翻转链表 —— 递归

25. K 个一组翻转链表

给你链表的头节点head，每k个节点一组进行翻转，请你返回修改后的链表。

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

你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

输入：head = [1,2,3,4,5], k = 2
输出：[2,1,4,3,5]

题目分析

按递归的方法，把整个链表划分为k个节点一组去处理，翻转前k个节点后，再对k+1节点后的链表进行递归处理，依次翻转链表。过程可以参考「206. 反转链表」

递归是一种算法结构。递归会出现在子程序中，形式上表现为直接或间接的自己调用自己

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
		if (head == null || head.next == null) {
			return head;
		}

		ListNode tail = head;
		// 剩余节点数不足k个直接返回
		for (int i = 0; i < k; i++) {
			if (tail == null) {
				return head;
			}
			tail = tail.next;
		}
		ListNode newHead = reverse(head, tail);
		head.next = reverseKGroup(tail, k);
		return newHead;
    }

	/**
	 * 翻转链表，指针翻转后，最终返回尾节点作为新链表的头结点
	 * @param head
	 * @param tail
	 * @return
	 */
	private ListNode reverse(ListNode head, ListNode tail) {
		ListNode pre = null, next = null;
		while (head != tail){
			next = head.next;
			head.next = pre;
			pre = head;
			head = next;
		}
		return pre;
	}
}