【卡码网C++基础课 14.链表的基础操作2】

---

题目描述与分析

题目描述：
请编写一个程序，实现以下操作：

构建一个单向链表，链表中包含一组整数数据，输出链表中的第 m 个元素（m 从 1 开始计数）。
要求：
1.使用自定义的链表数据结构
2.提供一个 linkedList 类来管理链表，包含构建链表、输出链表元素以及输出第 m 个元素的方法
3.在 main 函数中，创建一个包含一组整数数据的链表，然后输入 m，调用链表的方法输出第 m 个元素

输入描述：
第一行包含两个整数 n 和 k，n 表示需要构建的链表的长度，k 代表输入的 m 的个数。
接下来一行包含 n 个整数，表示链表中的元素。
接下来一行包含 k 个整数，表示输出链表中的第 m 个元素。

输出描述：
测试数据输出占 k 行。
每行输出链表中的第 m 个元素。如果 m 位置不合法，则输出“Output position out of bounds.”。

输入示例：

5 5
1 2 3 4 5
4 3 2 9 0

输出示例：

4
3
2
Output position out of bounds.
Output position out of bounds.

代码编写

依旧先把代码的基础结构给搭建好，先把链表结构体写出来：

#include <iostream>
using namespace std;
// 定义链表节点结构体
struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(nullptr) {}
};
int main() {
  
}

int n, k, m;
cin >> n >> k; // 输入n和k, n 表示需要构建的链表的长度，k 代表输入的 m 的个数。
ListNode* dummyHead = new ListNode(0); // 创建了一个虚拟头结点

和上一节内容相似，依旧需要使用for 循环，根据读取的 n 值，迭代读取 n 个整数，构建链表。

for (int i = 0; i < n; i++) { // 或者使用while(n--)
      int val; // 变量val接收输入的val值
    cin >> val;
    ListNode *newNode = new ListNode(val); // 构造一个新的节点
    cur -> next = newNode; // 将新节点接入链表
    cur = cur -> next;      // cur 指向下一个节点
}

上面的操作完成之后，链表就成功构建了，我们需要根据题目要求读取k个m，并输出这k个值

while(k--) {
  cin >> m;  // 输入k个m, m表示需要输出的节点的顺序
}

每次找到第m个节点都需要从头节点开始遍历，所以将 cur 重新指向虚拟头结点，以便重新遍历链表。

cur = dummyHead; // cur重新指向虚拟头节点，以便重新遍历链表

使用 while 循环，寻找链表中的第 m 个元素, 只要cur没有指向null,就一直循环下去

while(m--) {
  if(cur != NULL) {
    cur = cur->next;  // 找到第m个节点
  } else {
    break;
  }
}

当while循环结束后，如果cur指向null,说明输入的m已经超出了链表的长度，如果cur还是指向dummyNode说明没有进入到while循环，也就是m=0，这两种情况都是m位置不合法的情况。

// cur == NULL 表示 m 超出了链表长度
// cur == dummyHead 表示 m = 0
if (cur == NULL || cur == dummyHead) {
  cout << "Output position out of bounds." << endl;
} else {
  // cur表示第m个节点，正常输出cur节点的val值
  cout << cur->val << endl;
}

完整代码如下：

#include <iostream>
using namespace std;
struct ListNode {
    int val;
    ListNode* next;
    ListNode(int x) : val(x), next(nullptr) {}
};
int main() {
    int n, val, m, k;
    ListNode* dummyHead = new ListNode(0); // 创建了一个虚拟头结点
    cin >> n >> k;
    ListNode* cur = dummyHead; // 定义一个指向当前节点的指针 cur，初始指向虚拟头结点
    for (int i = 0; i < n; i++) { // 或者使用while(n--)
        cin >> val;
        ListNode* newNode = new ListNode(val); // 构造一个新的节点
        cur -> next = newNode; // 将新节点接入链表
        cur = cur -> next;      // cur 指向下一个节点
    }

    while (k--) {
        cin >> m;
        cur = dummyHead;
        while (m--) { // 寻找链表里第m个元素
            if(cur != NULL) cur = cur->next;
            else break;
        }
        // cur == NULL 表示 m 超出了链表长度
        // cur == dummyHead 表示 m = 0
        if (cur == NULL || cur == dummyHead) cout << "Output position out of bounds." << endl;
        else cout << cur->val << endl;
    }
}