Code Ganker: Rotate List -- LeetCode

2014年3月23日星期日

Rotate List -- LeetCode

原题链接: http://oj.leetcode.com/problems/rotate-list/
这是一道链表操作的题目,基本思路是用walker-runner定位到要旋转的那个结点,然后将下一个结点设为新表头,并且把当前结点设为表尾。需要注意的点就是旋转的结点数可能超过链表长度,所以我们要对这个进行取余。定位旋转的尾结点的不超过链表的长度,所以时间复杂度是O(n),空间复杂度是O(1)。代码如下:
public ListNode rotateRight(ListNode head, int n) {
    if(head == null)
    {
        return null;
    }
    ListNode walker = head;
    ListNode runner = head;
    int idx = 0;
    while(runner!=null && idx<n)
    {
        runner = runner.next;
        idx++;
    }
    if(runner == null)
    {
        n %= idx;
        runner = head;
        idx=0;
        while(runner!=null && idx<n)
        {
            runner = runner.next;
            idx++;
        }
    }
    while(runner.next!=null)
    {
        walker = walker.next;
        runner = runner.next;
    }
    runner.next = head;
    ListNode newHead = walker.next;
    walker.next = null;
    return newHead;
}
上面的实现中采取的方式是直接走到那个结点,如果没超过长度就直接旋转了,如果超过了,就进行取余,并且重新跑到结尾点。其实也可以先直接求长度,然后取余之后再走。其实各有利弊,所以大家根据喜好实现即可哈。

没有评论:

发表评论