华为-输出单向链表中倒数第k个结点

题目链接

​​https://www.nowcoder.com/practice/54404a78aec1435a81150f15f899417d?tpId=37&tqId=21274&tPage=3&rp=&ru=/ta/huawei&qru=/ta/huawei/question-ranking​​

题目描述

输入一个单向链表，输出该链表中倒数第k个结点，链表的倒数第1个结点为链表的尾指针。

链表结点定义如下：

struct ListNode

{

      int       m_nKey;

      ListNode* m_pNext;

};

详细描述：

接口说明

原型：

ListNode* FindKthToTail(ListNode* pListHead, unsignedint k);

输入参数：

        ListNode* pListHead  单向链表

     unsigned int k  倒数第k个结点

输出参数（指针指向的内存区域保证有效）：

    无

返回值：

        正常返回倒数第k个结点指针，异常返回空指针

输入描述:

输入说明
1 输入链表结点个数
2 输入链表的值
3 输入k的值

输出描述:

输出一个整数

输出一个整数

示例1

输入

复制

8
1 2 3 4 5 6 7 8
4

输出

复制

5

题解：

#include <iostream>
#include <string>
using namespace std;
struct node{
  int num;
  struct node *next;
};
int main(){
  int n;
  while (cin >> n){
    node *root = new node();
    node *p = root, *q = root;
    for (int i = 0; i < n; i++){
      int num;
      cin >> num;
      p->num = num;
      p->next = new node();
      p = p->next;
    }
    int k;
    cin >> k;
    if (k == 0){
      cout << "0" << endl;
    }
    else if (n < k){
      cout << "NULL" << endl;
    }
    else{
      for (int i = 0; i < n - k; i++){
        q = q->next;
      }
      cout << q->num << endl;
    }
  }
  return 0;
}