剑指offer:面试题22:链表中倒数第k个节点

题目:

输入一个链表,输出该链表中倒数第k个节点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾节点是倒数第1个节点。

例如,一个链表有 6 个节点,从头节点开始,它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

法1:最直接的寻找方式,遍历两遍链表。第一遍记录链表的长度,第二遍找到倒数第k个节点

倒数第k个节点为:n = n - k + 1;

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if(head == NULL) return head;
        int n = 0;
        ListNode *p = head;
        while(p != NULL)
        {
            n++;
            p = p->next;
        }
        n = n - k + 1;
        cout << n;
        if(n < 0) return NULL;
        p = head;
        while(n - 1)
        {
            p = p->next;
            n--;
        }
        return p;
    }
};

法2:只遍历一遍,定义一个前节点front,和后节点back

前节点先走k步,然后后节点和前节点一起往前走,当前节点为NULL时,即后节点为倒数第k个节点。

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        if(head == NULL) return head;
        ListNode *front = head;
        ListNode *behind = head;
        for(int i = 0; i < k; i++)
        {
            front = front->next;
            if(front == NULL) return head;
        }
        while(front != NULL)
        {
            front = front->next;
            behind = behind->next;
        }
        return behind;
    }
};

猜你喜欢

转载自blog.csdn.net/m0_63001277/article/details/127014978
今日推荐