链表中倒数第k个节点

目录

一、使用双指针

二、使用栈来存储节点

三、从控制台输入数字，创建链表并返回倒数第k个节点

---

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

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

示例：

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

一、使用双指针

设置两个节点，让第一个指针先走k-1，第二个指针再开始走，当第一个指针走到尾节点的时候，此时第二个指针就是倒数第k个。

class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {//双指针 第一个指针先走k-1步 第二个开始走 第一个走到末尾时的第二个指针就是倒数第k个节点
        if(head == nullptr || k == 0) return nullptr;
        
        ListNode* first = head;
        ListNode* last = nullptr;
        for(int kk = 0; kk < k - 1; ++kk){
            if(first->next == nullptr) return nullptr;//如果第一个指针到了链表结尾 但是k还没到1 说明k大于链表节点了
            first = first->next;
        }
        last = head;
        while(first->next != nullptr){
            last = last->next;
            first = first->next;
        }
        return last;
    }
};

二、使用栈来存储节点

 

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* getKthFromEnd(ListNode* head, int k) {
        stack<int> satcks ;
        ListNode*t =NULL;
       // ListNode*tmp = t;
        while(head){
            satcks.push(head->val);
            head = head->next;
        }
        while(k-->0 && !satcks.empty()){
            ListNode* rev = new ListNode();
            rev->val = satcks.top();
            satcks.pop();
            rev->next = t;
            t = rev;
        }
        return t;

    }
};

三、从控制台输入数字，创建链表并返回倒数第k个节点

​
#include <bits/stdc++.h>
using namespace std;

struct ListNode {
	int val;
	ListNode *next;
	ListNode(int x) : val(x), next(NULL) {}
	
};
class Solution {
public:
	ListNode* getKthFromEnd(ListNode* head, int k) {//双指针 第一个指针先走k-1步 第二个开始走 第一个走到末尾时的第二个指针就是倒数第k个节点
		if (head == nullptr || k == 0) return nullptr;

		ListNode* first = head;
		ListNode* last = nullptr;
		for (int kk = 0; kk < k - 1; ++kk) {
			if (first->next == nullptr) return nullptr;//如果第一个指针到了链表结尾 但是k还没到1 说明k大于链表节点了
			first = first->next;
		}
		last = head;
		while (first->next != nullptr) {
			last = last->next;
			first = first->next;
		}
		return last;
	}
};

int main()
{
	ListNode* head1 = new ListNode(NULL);
	ListNode* head = new ListNode(NULL);
	head1 = head;
	int a;
	while (cin >> a) {
		ListNode* tmp = new ListNode(a);
		head->next = tmp;
		head = tmp;
		if (cin.get() == '\n') break;
	}
	
	Solution aa;
	ListNode* pos1 = aa.getKthFromEnd(head1->next, 3);
	while (pos1->next != NULL) {
		cout << pos1->next->val << endl;
		pos1 = pos1->next;
	}
	
	system("pause");
	return 0;
}

​