LeetCode-61. 旋转链表-Java-medium

题目链接

法一

    /**
     * 思路：将尾部向前数第K个元素作为头，原来的头接到原来的尾上
     * 步骤：（1）先遍历求得链表总长度cnt，同时将链表首尾相连
     *      （2）再找到原链表的倒数第k+1个节点，该节点的next就是新链表的头结点
     *
     * @param head
     * @param k
     * @return
     */
    public ListNode rotateRight(ListNode head, int k) {
        if (head == null || head.next == null || k == 0) {
            return head;
        }
        int cnt = 1;
        ListNode cur = head;
        while (cur.next != null) {
            cnt++;
            cur = cur.next;
        }
        k %= cnt;   // 防止k超过链表长度
        if (k == 0) {
            return head;
        }
        cur.next = head; // 将链表首尾相连
        for (int i = 0; i < cnt - k; i++) { // 找到倒数第k+1个节点
            cur = cur.next;
        }
        ListNode newHead = cur.next; // 倒数第k+1个节点的next就是新链表的头结点
        cur.next = null; // 新链表的尾节点的next置为空
        return newHead;
    }

本地测试

        /**
         * 61. 旋转链表
         */
        lay.showTitle(61);
        Solution61 sol61 = new Solution61();
        ListNode head61 = new ListNode();
        int[] nums61 = new int[]{1, 2, 3, 4, 5};
        head61 = listOpt.creatListByArray(head61, nums61);
        listOpt.showList(head61);
        listOpt.showNoHeadList(sol61.rotateRight(head61.next, 2));