题解 | #链表中倒数最后k个结点#

链表中倒数最后k个结点

https://www.nowcoder.com/practice/886370fe658f41b498d40fb34ae76ff9

/**
 * struct ListNode {
 *	int val;
 *	struct ListNode *next;
 *	ListNode(int x) : val(x), next(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pHead ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    ListNode* FindKthToTail(ListNode* pHead, int k) {
        auto Size = [](ListNode* head) {
            size_t cnt{};
            for(auto cur = head; cur; cur = cur->next) {
                cnt ++;
            }
            return cnt;
        };
        return [](ListNode* head, size_t pos) {
            auto cur = head;
            while(cur and pos--) {
                cur = cur->next;
            }
            return cur;
        }(pHead, Size(pHead) - k);
    }
};

全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务