剑指offer面试题15：链表中倒数第k个结点

最新推荐文章于 2024-05-06 19:03:23 发布

原创最新推荐文章于 2024-05-06 19:03:23 发布 · 197 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#面试题

剑指offer 专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种高效查找链表中倒数第K个结点的方法，使用双指针技巧实现，适用于各类链表操作场景。

题目：输入一个链表，输出该链表中倒数第K个结点。链表的尾结点是倒数第1个节点。
思路：两个指针相差k-1。

#include <iostream>

using namespace std;

struct ListNode {
  int m_nValue;
  ListNode *m_pNext;
};

ListNode *CreateList() {
  int a;
  cin >> a;
  ListNode *head;
  head = NULL;
  if (a != -1) {
    head = new ListNode;
    head->m_nValue = a;
    head->m_pNext = CreateList();
  }
  return head;
}

ListNode *FindKthToTail(ListNode *phead, int k) {
  if (phead == NULL || k == 0) {
    cout << "Invalid Input!" << endl;
    return NULL;
  }
  ListNode *pFirst = phead;
  ListNode *pSecond = phead;
  for (int i = 1; i < k; i++) {
    if (pFirst->m_pNext != NULL)
      pFirst = pFirst->m_pNext;
    else {
      cout << "Invalid k!" << endl;
      return NULL;
    }
  }
  while (pFirst->m_pNext != NULL) {
    pFirst = pFirst->m_pNext;
    pSecond = pSecond->m_pNext;
  }
  cout << pSecond->m_nValue << endl;
  return pSecond;
}

int main() {
  int k = 1;
  ListNode *head = CreateList();
  ListNode *Kth = FindKthToTail(head, k);
  (Kth == NULL) ? cout << "Invalid!" << endl : cout << Kth->m_nValue << endl;
  return 0;
}