leetcode25 Reverse Nodes in k-Group

Reverse Nodes in k-Group

My Submissions

Question

Total Accepted: 47206  Total Submissions: 181287  Difficulty: Hard

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




/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public class Solution {
    public static ListNode reverse(ListNode p, int k) {
		ListNode pre = new ListNode(-1);
		ListNode px = p;
		ListNode q = px.next;
		for (int i = 0; i < k ; i++) {
			q = px.next;
			px.next = pre.next;
			pre.next = px;
			px = q;
		}
		return pre.next;
	}
    public ListNode reverseKGroup(ListNode head, int k) {
        if(head==null || head.next==null || k<1){
			return head;
		}
		ListNode pre = new ListNode(-1);
		ListNode startpre = pre;
	    ListNode p = head;
	    //ListNode prem = pre;
	    ListNode pm = p;
	   // boolean flag = false;
	    ListNode r = head;
	    int count = 0;
	    while(r!=null){
	    	r = r.next;
	    	count++;
	    }
	    if(count<k){
	    	return head;
	    }
	    while(true){
	    	int i;
	    	for( i=0;i<k && pm!=null;i++){
	    		//prem = pm;
	    		pm = pm.next;
	    	}
	    	/*if(pm==null && i<k &&flag == false){
	    		return head;
	    	}
	    	if(pm==null && i>=k && flag == false){
	    		ListNode x = reverse(p, k);
	    		pre.next = x;
	    	}*/
	    	if(i>=k ){
	    	//	flag = true;
	    		ListNode x = reverse(p, k);
	    		pre.next = x;
	    		/*pre = prem;*/
	    		for(int j=0;j<k;j++){
	    			pre = pre.next;
	    		}
	    		pre.next = pm;
	    		p = pm;
	    	}
	    	if(i<k){
	    		
	    		break;
	    	}
	    }
	    return startpre.next;
    }
}

可运行代码




package lianbiaofanzhuan;

public class Solution {
	public static ListNode create(int[] a) {
		ListNode head = new ListNode(a[0]);
		ListNode p = head;
		for (int i = 1; i < a.length; i++) {
			ListNode q = new ListNode(a[i]);
			p.next = q;
			p = q;
		}
		return head;
	}

	public static void display(ListNode head) {
		ListNode p = head;
		while (p != null) {
			System.out.println(p.val);
			p = p.next;
		}
	}

	public static ListNode reverse(ListNode head) {
		ListNode pre = new ListNode(-1);
		ListNode p = head;
		ListNode q;
		while (p != null) {
			q = p.next;
			p.next = pre.next;
			pre.next = p;
			p = q;
		}
		return pre.next;
	}

	public static ListNode reverse(ListNode p, int k) {
		ListNode pre = new ListNode(-1);
		ListNode px = p;
		ListNode q = px.next;
		for (int i = 0; i < k ; i++) {
			q = px.next;
			px.next = pre.next;
			pre.next = px;
			px = q;
		}
		return pre.next;
	}
	public static ListNode reverseKGroup(ListNode head, int k) {
		if(head==null || head.next==null || k<1){
			return head;
		}
		ListNode pre = new ListNode(-1);
		ListNode startpre = pre;
	    ListNode p = head;
	    //ListNode prem = pre;
	    ListNode pm = p;
	   // boolean flag = false;
	    ListNode r = head;
	    int count = 0;
	    while(r!=null){
	    	r = r.next;
	    	count++;
	    }
	    if(count<k){
	    	return head;
	    }
	    while(true){
	    	int i;
	    	for( i=0;i<k && pm!=null;i++){
	    		//prem = pm;
	    		pm = pm.next;
	    	}
	    	/*if(pm==null && i<k &&flag == false){
	    		return head;
	    	}
	    	if(pm==null && i>=k && flag == false){
	    		ListNode x = reverse(p, k);
	    		pre.next = x;
	    	}*/
	    	if(i>=k ){
	    	//	flag = true;
	    		ListNode x = reverse(p, k);
	    		pre.next = x;
	    		/*pre = prem;*/
	    		for(int j=0;j<k;j++){
	    			pre = pre.next;
	    		}
	    		pre.next = pm;
	    		p = pm;
	    	}
	    	if(i<k){
	    		
	    		break;
	    	}
	    }
	    return startpre.next;
    }
    
	public static void main(String[] args) {
		int[] a = { 1, 2};
		ListNode head = create(a);
		//display(head);
		/*ListNode headB = reverse(head);
		System.out.println();
		display(headB);*/
		
		
		ListNode headC = reverseKGroup(head, 3);
		display(headC);
	}

}

class ListNode {
	int val;
	ListNode next;

	ListNode(int x) {
		val = x;
	}
}