lsmCC 发表于 2024-1-7 19:12:37

修改代码

void deleteEmployee() {
    int id;
    printf("请输入要删除的员工工号:");
    scanf("%d", &id);

    Employee* current = head;
    Employee* prev = NULL;
    Employee* temp = NULL;

    while (current != NULL) {
      if (current->id == id) {
            if (prev == NULL) {
                head = current->next;
            } else {
                prev->next = current->next;
            }
            temp = current;
            current = current->next;
            free(temp);// 释放删除的节点
            printf("员工信息删除成功!\n");
            break;
      }
      prev = current;
      current = current->next;
    }

    _save(head);// 将更新后的链表保存到文件中
}
如果员工只有一个人无法删除,怎么修改

FishC 发表于 2024-1-7 19:12:49

如果链表中只有一个员工,无法进行删除操作。你可以在删除之前先判断链表中的节点个数,如果只有一个节点,则输出相应提示信息,不执行删除操作。

修改后的代码如下所示:

void deleteEmployee() {
    int id;
    printf("请输入要删除的员工工号:");
    scanf("%d", &id);

    Employee* current = head;
    Employee* prev = NULL;
    Employee* temp = NULL;

    // 判断链表中的节点个数
    int count = 0;
    while (current != NULL) {
      count++;
      current = current->next;
    }
   
    if (count == 1) {
      printf("当前链表中只有一个员工,无法进行删除操作!\n");
      return;
    }

    current = head;// 重新将指针指向链表头部

    while (current != NULL) {
      if (current->id == id) {
            if (prev == NULL) {
                head = current->next;
            } else {
                prev->next = current->next;
            }
            temp = current;
            current = current->next;
            free(temp);// 释放删除的节点
            printf("员工信息删除成功!\n");
            break;
      }
      prev = current;
      current = current->next;
    }

    _save(head);// 将更新后的链表保存到文件中
}

修改后的代码会先检查链表中的节点个数,如果只有一个节点,则直接输出提示信息并返回,不进行删除操作。否则,继续按照原有逻辑执行删除操作。

以上回复来自 FishC AI 小助理,如未能正确解答您的问题,请继续追问。
页: [1]
查看完整版本: 修改代码