`
frank-liu
  • 浏览: 1666844 次
  • 性别: Icon_minigender_1
  • 来自: 北京
社区版块
存档分类
最新评论

leetcode: Rotate List

 
阅读更多

问题描述:

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

For example:
Given 1->2->3->4->5->NULL and k = 2,
return 4->5->1->2->3->NULL.

原问题链接:https://leetcode.com/problems/rotate-list/

 

问题分析

    对于这个问题来说,它本身很容易找到思路去解决。主要是实现容易出现各种小的错误。因为要将整个链表循环向右移动k位,所以我们就需要找到从链表的最后面往前的k个元素。因为按照最终的调整,这个位置的元素将作为新链表的结尾,而它后面的那个节点则将作为新链表的开头。

  那么,我们该怎么来实现呢?从具体实现的步骤来看,我们需要找到倒数第k个元素,同时要找到它前面一个的元素,来进行操作。另外,既然是循环移位,原来链表最末尾的元素要和原来的链表头连接起来。另外一个,对于参数k来说,如果它本身比较大的话,我们就这么去循环的找也不是个办法。我们可以通过求得链表的总长度,再将k对链表长度取模,这样就可以减少遍历的距离。如果按照这种方式实现的话,我们需要有两个指针,一个遍历到链表最后将整个链表给连起来。另外一个记录倒数第k个元素的位置,然后将它前面的元素和它断开。

  结合前面我们需要计算链表的长度来对k取模,我们可以将整个过程简化成只需要一个指针。首先我们用一个指针遍历到链表的末尾,并同时计算链表的长度。这样我们将k对链表长度取模之后再往前移动n - k 个位置。这个时候这个位置就是我们需要断开的节点,我们将这个节点断开,并返回它后面的元素作为新链表的头,则得到我们期望的结果。详细的代码实现如下:

 

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(head == null || head.next == null || k == 0) return head;
        ListNode p = head;
        int len = 1;
        while(p.next != null) {
            p = p.next;
            len++;
        }
        p.next = head;
        k %= len;
        for(int i = 0; i < len - k; i++) p = p.next;
        head = p.next;
        p.next = null;
        return head;
    }
}

 

 

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics