随笔-旋转链表

版权声明:中华人民共和国持有版权 https://blog.csdn.net/Fly_Fly_Zhang/article/details/85239368

题目:

给定一个链表,旋转链表,将链表每个节点向右移动 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

思路:先确定链表多长,然后%出它到底要右移几位。在找到当头结点的前一项;

这里面唯一要注意的是 head 也参与循环。并且交换完成后,一定要将头结点的pre 置为空,要不然就变成环

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(k==0 ||head==null || head.next==null){
            return head;
        }
        int len=0;
        ListNode cur=head; //链表最后一个元素
        while(cur.next!=null){
            len++;
            cur=cur.next;
        }
        len++;
        k=k%len;
        ListNode pre=head;
        len=len-k;
        while(len>1){
            pre=pre.next;
            len--;
        }
        cur.next=head;
        cur=pre.next;
        pre.next=null;
        return cur;
    }
}

猜你喜欢

转载自blog.csdn.net/Fly_Fly_Zhang/article/details/85239368