leetcode-025-k个一组翻转链表

题目及测试

package pid025;
/* 25. K 个一组翻转链表
给你链表的头节点 head ，每 k 个节点一组进行翻转，请你返回修改后的链表。

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

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

 

示例 1：


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



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

提示：
链表中的节点数目为 n
1 <= k <= n <= 5000
0 <= Node.val <= 1000
 

进阶：你可以设计一个只用 O(1) 额外内存空间的算法解决此问题吗？

*/



public class main {
	
	
	public static void main(String[] args) {
		
		LinkList a=new LinkList(1);
		a.addLast(2);
		a.addLast(3);
		a.addLast(4);
		a.addLast(5);
		
		test(a.first,2);
		
		a=new LinkList(1);
		a.addLast(2);
		a.addLast(3);
		a.addLast(4);
		a.addLast(5);
		
		
		
		
		test(a.first,3);
	}
		 


	private static void test(ListNode ito,int ito2) {
		Solution solution = new Solution();
		ListNode rtn;
		long begin = System.currentTimeMillis();
		System.out.println();
		//开始时打印数组
		ito.printNodeToEnd();
		System.out.println("ito2="+ito2);
		rtn=solution.reverseKGroup(ito,ito2);//执行程序
		rtn.printNodeToEnd();
		long end = System.currentTimeMillis();	
		
		//System.out.println(":rtn" );
		//System.out.print(rtn);
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

解法1（成功，0ms，极快）

步骤分解:

链表分区为已翻转部分+待翻转部分+未翻转部分
每次翻转前，要确定翻转链表的范围，这个必须通过 k 此循环来确定
需记录翻转链表前驱和后继，方便翻转完成后把已翻转部分和未翻转部分连接起来
初始需要两个变量 pre 和 end，pre 代表待翻转链表的前驱，end 代表待翻转链表的末尾
经过k此循环，end 到达末尾，记录待翻转链表的后继 next = end.next
翻转链表，然后将三部分链表连接起来，然后重置 pre 和 end 指针，然后进入下一次循环
特殊情况，当翻转部分长度不足 k 时，在定位 end 完成后，end==null，已经到达末尾，说明题目已完成，直接返回即可
时间复杂度为 O(n*K) 最好的情况为 O(n)最差的情况未 O(n^2)

package pid025;

import java.nio.channels.NonWritableChannelException;

/**
 * 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 || k<=1){
    		return head;
    	}
    	ListNode resultHead = new ListNode();
    	
    	ListNode nowSegHead = head;
    	ListNode nowSegTail = null;
    	ListNode nextSegHead = null;
    	ListNode prevSegTail = resultHead;
    	boolean end = false;
    	
    	ListNode now = null;
    	ListNode nowNext = null;
    	ListNode nowNext2 = null;
    	while(true){
    		now = nowSegHead;
    		// 如果now是null，说明直接没有这轮了，上一轮的tail的next已经指向next，无需处理
    		if(now == null){
    			break;
    		}
    		// 从now走k-1步，如果有null，说明是最后一段
    		for(int i=0;i<k-1;i++){
    			now = now.next;
    			if(now == null){
    				end = true;
    				break;
    			}
    		}
    		//如果是最后一段，处理上轮tail的next，结束
    		if(end){
    			prevSegTail.next = nowSegHead;
    			break;
    		}else{
    			//中间的段，设置这轮的tail和下轮的head
    			nowSegTail = now;
    			nextSegHead = now.next;
    		}
    		//得到一个完整的段，翻转
    		now = nowSegHead;
    		nowNext = now.next;
    		for(int i=0;i<k-1;i++){
    			nowNext2 = nowNext.next;
    			nowNext.next = now;
    			now = nowNext;
    			nowNext = nowNext2;
    		}
    		// 处理这轮的head和tail的next
    		nowSegHead.next = null;
    		prevSegTail.next = nowSegTail;
    		// 进入下轮，上轮的tail的nowHead，这轮的head是之前的nextHead
    		prevSegTail = nowSegHead;
    		nowSegHead = nextSegHead;
    	}
    	return resultHead.next;
    }
}

解法2（别人的）

public ListNode reverseKGroup(ListNode head, int k) {
    ListNode dummy = new ListNode(0);
    dummy.next = head;

    ListNode pre = dummy;
    ListNode end = dummy;

    while (end.next != null) {
        for (int i = 0; i < k && end != null; i++) end = end.next;
        if (end == null) break;
        ListNode start = pre.next;
        ListNode next = end.next;
        end.next = null;
        pre.next = reverse(start);
        start.next = next;
        pre = start;

        end = pre;
    }
    return dummy.next;
}

private ListNode reverse(ListNode head) {
    ListNode pre = null;
    ListNode curr = head;
    while (curr != null) {
        ListNode next = curr.next;
        curr.next = pre;
        pre = curr;
        curr = next;
    }
    return pre;
}