[程序员代码面试指南]链表问题-将单链表的每k个节点之间逆序

题目描述

给定一个单链表的表头节点head，实现一个调整单链表的函数，是的每k个节点之间逆序，如果最后不够k个节点一组，则不调整最后几个节点。

题解

内部函数reverse实现链表beg到end的翻转，以及与l和r的连接。

外部函数reverseKNodes实现区间的移动(即四个实参的更新)，以及整个链表头节点的赋值。

时间复杂度O(n),额外空间复杂度O(1)

代码

public class Main {
	public static void main(String args[]) {
		Node n1=new Node(1);
		Node n2=new Node(2);
		Node n3=new Node(3);
		Node n4=new Node(4);
		Node n5=new Node(5);
		n1.next=n2;
		n2.next=n3;
		n3.next=n4;
		n4.next=n5;

		Node head=n1;
		int k=2;
		Node newHead=reverseKNodes(head,k);
		Node p=newHead;
		while(p.next!=null) {
			System.out.println(p.val);
			p=p.next;
		}
	}

	public static Node reverseKNodes(Node head,int k) {
		if(k<2) {
			return head;
		}

		boolean firstKNodesTag=true;//是否是第一个小组
		Node newHead=null;//新链表表头

		Node l=null;//前一个小组的尾节点
		Node beg=null;//小组开始节点
		Node cur=head;//当前
		Node next=null;//当前下一个节点

		int cnt=1;

		while(cur!=null) {
			next=cur.next;
			if(cnt==k) {
				if(firstKNodesTag) {//新链表表头
					newHead=cur;
					beg=head;
					firstKNodesTag=false;
				}

				reverse(l,beg,cur,next);
				beg=next;//
				l=beg;
				cnt=0;
			}
			cur=next;
			++cnt;
		}
		return newHead;
	}

	public static void reverse(Node l,Node beg,Node end,Node r) {
		Node pre=beg;//
		Node cur=beg.next;//
		Node next=null;
		while(cur!=r) {
			next=cur.next;
			cur.next=pre;
			pre=cur;
			cur=cur.next;
		}
		if(l!=null) {
			l.next=end;
		}
		beg.next=r;//
	}
}