卡码网语言基础课 ｜链表的基础操作II

题目分析

需要注意的点有：

• 需要考虑 m 是否在链表元素的范围内，有两种情况：
  第一，m 超出元素范围：例如 m = 5，链表Link = {1, 2, 3}，
  第二，m = 0，同样不在链表范围内
• 因为要输出不止一个 m ，所以每当找到第 m 个点后，都应该将 cur 重新定位到虚拟头节点，以便继续进行寻找 m

代码编写

首先实现代码的基础结构

#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++) {
	int 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 {
	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++) {
        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;
        } 
            
    }
}