码迷,mamicode.com
首页 > 其他好文 > 详细

【LeetCode】Reverse Nodes in k-Group

时间:2014-12-12 19:06:05      阅读:133      评论:0      收藏:0      [点我收藏+]

标签:链表

题目

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

解答

题目要求对链表的每K个元素,翻转一次。最后一次如果不到K,则不用翻转。

可以通过链表的区间内翻转实现,遇到k组的节点就把前k个节点翻转,代码如下:

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public class Solution {
    public static ListNode reverseKGroup(ListNode head,int k){
    	if(head==null||k==1){
    		return head;
    	}
    	ListNode fakeHead=new ListNode(-1);
    	fakeHead.next=head;
    	ListNode pre=fakeHead;
    	int i=0;
    	while(head!=null){
    		i++;
    		if(i%k==0){
    			pre=reverse(pre,head.next);  //reverse返回的last就是pre
    			head=pre.next;
    		}else{
    			head=head.next;
    		}
    	}
    	return fakeHead.next;
    }

    private static ListNode reverse(ListNode pre,ListNode next){  //区间的链表反转
    	ListNode last=pre.next;  //last不变
    	ListNode cur=last.next;  //cur移动
    	while(cur!=next){
    		last.next=cur.next;
    		cur.next=pre.next;
    		pre.next=cur;
    		cur=last.next;
    	}
    	return last;
    }
}


---EOF---

【LeetCode】Reverse Nodes in k-Group

标签:链表

原文地址:http://blog.csdn.net/navyifanr/article/details/41897455

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!