卡码网编程基础课 | 链表的基础操作II

问题描述

题目描述

请编写一个程序，实现以下操作： 

构建一个单向链表，链表中包含一组整数数据，输出链表中的第 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.

python解法

# 定义链表节点
class Node:
    def __init__(self, data):
        self.data = data
        self.next = None
        
# 定义链表
class LinkList:
    def __init__(self):
        self.head_node = None
        self.length = 0
        
    def insert(self, data):
        self.length += 1
        new_node = Node(data)
        if self.head_node is None:
            self.head_node = new_node
            return self.head_node
        else:
            current_node = self.head_node
            while current_node.next is not None:
                current_node = current_node.next
            current_node.next = new_node
            return new_node
    # get方法用于查找 第 n 个节点
    def get(self, n):
        # 如果 n 小于 1 或者 大于链表的长度，返回 None
        if n < 1 or n > self.length:
            return None
        i = 1
        # 从头开始遍历
        current_node = self.head_node
        while current_node is not None:
            if i == n:  # 当 i == n 的时候,找到第 n 个节点
                return current_node
            i += 1
            current_node = current_node.next    # 移动 current_node指针,转为下一个节点
        return None
            
n, k = map(int, input().split())
# 新建链表实例
link_list = LinkList()
elements = list(map(int, input().split()))
# 将输入数据插入到链表中
for data in elements:
    link_list.insert(data)
# 获取输入的m
m_values = list(map(int, input().split()))

for m in m_values:
    node = link_list.get(m)
    if node is not None:
        print(node.data)
    else:
        print("Output position out of bounds.")

C++解法

# include<iostream>

// 定义链表结构体
struct ListNode{
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(nullptr) {}
};

int main(){
    int n, k, m;
    std::cin >> n >> k; // 获取输入n和k,n表示要构建的链表长度,k表示输入的m的个数
    ListNode *dummyHead = new ListNode(0);  // 创建了一个虚拟头节点
    ListNode *cur = dummyHead;  // 创建cur来指向虚拟头节点
    // 将读取到的值放入链表中
    for (int i = 0; i < n; i++){
        int val;
        std::cin >> val;
        ListNode *newNode = new ListNode(val);
        cur->next = newNode;
        cur = cur->next;
    }
    
    // 寻找第m个元素
    while (k--){
        std::cin >> m;
        cur = dummyHead;
        while (m--){
            if (cur != NULL) cur = cur->next;
            else break;
        }
        // cur == NULL 表示 m 超出了链表长度
        // cur == dummyHead 表示 m = 0
        if (cur == NULL || cur == dummyHead)
            std::cout << "Output position out of bounds." << std::endl;
        else
            std::cout << cur->val << std::endl;
    }
}