[数据结构实验]学生成绩管理

#include <iostream>
#include <cstdio>
#include <malloc.h>
#include <cstring>
#define NULL 0
#define LEN sizeof(Node)
using namespace std;
const char lesson[][12] = {"Math", "English", "Chinese", "physics", "Chemistry"};
struct Node {
        char ID[110];
        int arr[5];
        struct Node *next;
        Node() {
                memset(ID, 0, sizeof(ID));
                memset(arr, 0, sizeof(arr));
                next = NULL;
        }
};

struct StudentGradeManageSystem {
        Node ListHead;
        Node *head = &ListHead;
        bool find(char str[]) {
                Node *p = ListHead.next;
                while(p) {
                        if(strcmp(p->ID, str) == 0) return 1;
                        p = p->next;
                }
                return 0;
        }
        // add students' information
        void addStu() {
                Node *p = head;
                while(p->next != NULL) p = p->next;
                while(1) {
                        Node *tmp;
                        tmp = (Node*) malloc(LEN);
                        tmp->next = NULL;
                        cout<< endl;
                        puts("*************** add student information ****************");
                        cout<< "Input ID(input 0 to exit): ";
                        cin>> tmp->ID;
                        if(find(tmp->ID)) {
                                printf("ID already exist!
");
                                continue;
                        }
                        if(strlen(tmp->ID) == 1 && tmp->ID[0] == '0') break;
                        for(int i = 0; i < 5; i++) {
                                printf("Input %s grade: ", lesson[i]);
                                cin>> tmp->arr[i];
                        }
                        p = p->next = tmp;
                        cout<< endl;
                }
        }
        // delete students' information
        void deleteStu() {
                while(1) {
                        char str[102];
                        cout<< endl;
                        puts("*************** delete student information ****************");
                        cout<< "Input ID(input 0 to exit): ";
                        cin>> str;
                        if(strlen(str) == 1 && str[0] == '0') break;
                        Node *p = head->next, *fa = head;
                        int F = 0;
                        while(p) {
                                if(strcmp(p->ID, str) == 0) {
                                        fa->next = p->next;
                                        F = 1;
                                        break;
                                }
                                fa = p;
                                p = p->next;
                        }
                        if(F) puts("successfully delete!");
                        else puts("ID don't exist!");
                        cout<< endl;
                }
        }
        // view students' information
        void displayStu() {
                cout<< endl;
                cout<< "************************* view student information ****************************"<< endl;
                printf("%12s", "ID");
                for(int i = 0; i < 5; i++) printf("%12s", lesson[i]);
                cout<< endl;
                Node *p = head->next;
                while(p) {
                        printf("%12s", p->ID);
                        for(int i = 0; i < 5; i++) printf("%12d", p->arr[i]);
                        cout<< endl;
                        p = p->next;
                }
                cout<< "*******************************************************************************"<< endl;
        }
};
int Menu()
{
        cout<< endl;
        puts("---------------- Welcome to Student Grade Manage System --------------");
        puts("0: exit");
        puts("1: add students' information");
        puts("2: delete students' information");
        puts("3: view students' information");
        printf("Input: ");
        int x;
        cin>> x;
        return x;
}
int main()
{
        StudentGradeManageSystem G;
        while(1) {
                int key = Menu();
                if(key == 0) break;
                if(key == 1) G.addStu();
                if(key == 2) G.deleteStu();
                if(key == 3) G.displayStu();
        }
        return 0;
}