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 ->

结论

如果您对本教程有任何疑问,请在评论部分提出。

更新于:2020年12月30日

浏览量:115

开启您的职业生涯

完成课程获得认证

开始学习
广告