C++单链表中删除所有非素数节点
在本教程中,我们将学习如何从单链表中删除所有素数节点。
让我们看看解决问题的步骤。
编写包含数据和下一个指针的结构体。
编写一个函数将节点插入到单链表中。
用虚拟数据初始化单链表。
遍历单链表。查找当前节点数据是否为素数。
如果当前数据不是素数,则删除该节点。
编写一个删除节点的函数。删除节点时,请考虑以下三种情况。
如果节点是头节点,则将头节点移动到下一个节点。
如果节点是中间节点,则将下一个节点链接到前一个节点。
如果节点是尾节点,则删除前一个节点的链接。
示例
让我们看看代码。
#include <bits/stdc++.h> using namespace std; struct Node { int data; Node* next; }; void insertNode(Node** head_ref, int new_data) { Node* new_node = new Node; new_node->data = new_data; new_node->next = (*head_ref); (*head_ref) = new_node; } bool isPrime(int n) { if (n <= 1) { return false; } if (n <= 3) { return true; } if (n % 2 == 0 || n % 3 == 0) { return false; } for (int i = 5; i * i <= n; i = i + 6) { if (n % i == 0 || n % (i + 2) == 0) { return false; } } return true; } void deleteNonPrimeNodes(Node** head_ref) { Node* ptr = *head_ref; while (ptr != NULL && !isPrime(ptr->data)) { Node *temp = ptr; ptr = ptr->next; delete(temp); } *head_ref = ptr; if (ptr == NULL) { return; } Node *curr = ptr->next; while (curr != NULL) { if (!isPrime(curr->data)) { ptr->next = curr->next; delete(curr); curr = ptr->next; } else { ptr = curr; curr = curr->next; } } } void printLinkedList(Node* head) { while (head != NULL) { cout << head->data << " -> "; head = head->next; } } int main() { Node* head = NULL; insertNode(&head, 1); insertNode(&head, 2); insertNode(&head, 3); insertNode(&head, 4); insertNode(&head, 5); insertNode(&head, 6); cout << "Linked List before deletion:" << endl; printLinkedList(head); deleteNonPrimeNodes(&head); cout << "\nLinked List after deletion:" << endl; printLinkedList(head); }
输出
如果运行以上代码,则会得到以下结果。
Linked List before deletion: 6 -> 5 -> 4 -> 3 -> 2 -> 1 -> Linked List after deletion: 5 -> 3 -> 2 ->
结论
如果您对本教程有任何疑问,请在评论部分提出。
广告