题解 | #调整牛群顺序#

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param head ListNode类 
     * @param n int整型 
     * @return ListNode类
     */
    ListNode* moveNthToEnd(ListNode* head, int n) {
        // write code here

        // 确定head的长度
        ListNode* temp = head;
        int len = 0;
        while(temp)
        {
            ++len;
            temp = temp->next;
        }


        ListNode* ans = new ListNode(-1);
        ans->next = head;
        temp = ans;
        int num=-1;
        ListNode* t_next = nullptr;
        while(temp && temp->next)
        {
            ++num;
            // 到达指定位置的前一位
            if(num == len-n)
            {
                // 注意此时t_next可能不止有一个节点哦，可能是一段子链表
                t_next = temp->next;
                // temp->next->next有可能为空;
                temp->next = temp->next->next;
            }

            // 确保temp不为空
            if(temp->next)
                temp = temp->next;
        }

        temp->next = t_next;
        temp = temp->next;

        if(temp && temp->next)
            temp->next = nullptr;

        return ans->next;
    }
};