[leetcode] Rotate List

Given a list, rotate the list to the right by k places, where k is non-negative.

For example:
Given1->2->3->4->5->NULLand k =2,
return4->5->1->2->3->NULL.

https://oj.leetcode.com/problems/rotate-list/

思路1:先遍历一遍求长度,然后从头再开始走k%len步,设置新的头尾节点,连接原来的头尾节点。

思路2:遍历求长度,将首尾连接,继续走到新的头尾,断开。

 

思路1代码:

public class Solution {
    public ListNode rotateRight(ListNode head, int n) {
        if (head == null)
            return null;
        int len = 0;
        ListNode p = head;
        while (p != null) {
            len++;
            p = p.next;
        }
        
        p = head;
        int i;
        n = n % len;
        if (n == 0)
            return head;
        for (i = 0; i < len - n - 1; i++)
            p = p.next;
        ListNode newTail = p;
        ListNode newHead = p.next;

        while (p.next != null)
            p = p.next;
        p.next = head;
        newTail.next = null;
        return newHead;

    }

    public static void main(String[] args) {
        ListNode head = ListUtils.makeList(1, 2, 3, 4, 5);
        ListUtils.printList(head);
        head = new Solution().rotateRight(head, 2);
        ListUtils.printList(head);
    }
}

 

第二遍记录:

  改了个小地方,第一遍求长度的时候,顺便把头尾连接起来。

public class Solution {
    public ListNode rotateRight(ListNode head, int n) {
        if(head==null)
            return null;
        int len=1;
        ListNode p = head;
        while(p.next!=null){
            len++;
            p=p.next;
        }
        p.next=head;
        
        n = n%len;
        p=head;
        for(int i=0;i<len-1-n;i++){
            p=p.next;
        }
        ListNode newHead =p.next;
        p.next=null;
        return newHead;
    }
}

 

 

posted @ 2014-06-26 20:11  jdflyfly  阅读(177)  评论(0编辑  收藏  举报