给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。

示例 1:

输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
示例 2:

输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL

链表比较麻烦的一点是,只能单向走,这题我的思路是先确定链表的长度,确定要动的点。

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

答案比较取巧,首先把他闭合成环,然后确定位置断开。

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

程浩
21 声望2 粉丝

« 上一篇
60. 第k个排列
下一篇 »
62. 不同路径