【链表+内存池】学生管理系统

#include<iostream>
#include<iomanip>

using namespace std;


typedef struct _DATA_
{
    int iAge;
    int iGrade;
    char szName[20];
}Data,*pData;



typedef struct _NODE_
{
    Data DataTemp;
    _NODE_* pNext;
    
}Node,*pNode;


class CList
{
public:
    CList()
    {
        m_pHead = m_pTail = NULL;

        m_iNodeCount = 0;
    }
    ~CList()
    {

    }

    pNode CreateNode(Data DataTemp, CList& CFreeListObj);
    void LinkNode(pNode pNodeTemp);

    void RemoveNode(pNode pNodeTemp,CList& CFreeListObj);

    int GetNodeCount();

    bool RecycleNode(pNode pNodeTemp);

    pNode AllocateNode();
        
    void InitList()
    {
        m_pHead = m_pTail = NULL;
        m_iNodeCount = 0;

    }
    
    bool DestroyList();
    void TravelList()
    {
        pNode pNodeTemp = m_pHead;
        while(pNodeTemp != NULL)
        {
            cout<<"Name: "<<setw(12)
                <<pNodeTemp->DataTemp.szName<<"  "
                <<"Age: "<<pNodeTemp->DataTemp.iAge<<endl;
            pNodeTemp = pNodeTemp->pNext;
        
        }
    }

/*    void operator = (CList& CListTemp)
    {
        this->m_pHead = CListTemp.m_pHead;
        this->m_pTail = CListTemp.m_pTail;
        this->m_iNodeCount = CListTemp.m_iNodeCount;
    }
*/
    friend class CFile;

private:
    pNode m_pHead;
    pNode m_pTail;

    int m_iNodeCount;
};



class CFile :public CList
{
public:

    FILE* m_fp;
    char m_szFileRoute[256];

    CFile()
    {
        m_fp = NULL;

        memset(m_szFileRoute,0,sizeof(char)*256);
    }

    int  GetFileLen();

    pNode GetHead()
    {
        return m_pHead;
    }


    void GetFileRoute(char* szRoute);

    void FileRead(char* szRoute,CList& CFreeListObj);

    void FileWrite(char* szRoute,CList& CFreeListObj);

protected:
private:
};

int CFile::GetFileLen()
{
    
    int nLen = 0;
    
    fseek(m_fp,0L,SEEK_END);   
    nLen = ftell(m_fp);   // 获得文件大小
    rewind(m_fp);    //再将文件流指针定位开头

    return nLen;
}

void CFile::GetFileRoute(char* szRoute)
{
    strcpy(m_szFileRoute,szRoute);
}


void CFile::FileRead(char* szRoute,CList& CFreeListObj)
{
    GetFileRoute(szRoute);
    
    m_fp = fopen(m_szFileRoute,"r");
    
    if (m_fp==NULL)
    {
        m_fp = fopen(m_szFileRoute,"w");

        fclose(m_fp);


        return;
        
        
    }
    else  // 代表文件存在
    {
        
        int nLen = GetFileLen();
        
        if (nLen==0)  // 文件中没有数据
        {
            
            fclose(m_fp);

            return;
        }
        
        else   // 文件有数据
        {
        

            //读

            int i = 0;

            Data DataTemp = {0};
            pNode pNodeTemp = NULL;
            for (i=0;i<nLen/sizeof(Data);i++)
            {

                fread(&DataTemp,sizeof(Data),1,m_fp);

                pNodeTemp = CreateNode(DataTemp,CFreeListObj);

                LinkNode(pNodeTemp);

            }
                
            fclose(m_fp);

            return;
            
        }
        
    }
}

void CFile::FileWrite(char* szRoute,CList& CFreeListObj)
{

    GetFileRoute(szRoute);
    m_fp = fopen(m_szFileRoute,"w");
    pNode pNodeHead = GetHead();
    pNode pNodeTemp = pNodeHead;


    while (pNodeTemp!=NULL)
    {
        fwrite(&(pNodeTemp->DataTemp),sizeof(Data),1,m_fp);

        pNodeTemp = pNodeTemp->pNext;
    }


    fclose(m_fp);
}




pNode CList::CreateNode(Data DataTemp, CList& CFreeListObj)
{
    pNode pNodeTemp = NULL;

    int iNodeCount = CFreeListObj.GetNodeCount();

    if(iNodeCount > 0)
    {
        pNodeTemp = CFreeListObj.AllocateNode();
        cout<<"FreeList."<<endl;

    }
    else
    {
        pNodeTemp = new Node;
    //    cout<<"New."<<endl;
    }


    if(pNodeTemp != NULL)
    {
        pNodeTemp->DataTemp = DataTemp;
        pNodeTemp->pNext = NULL;


        m_iNodeCount++;

        return pNodeTemp;
    }
    else
    {
        return NULL;
    }        
}


void CList::LinkNode(pNode pNodeTemp)
{

    if(m_pHead == NULL)
    {
        m_pHead = m_pTail = pNodeTemp;
    }
    else
    {
        m_pTail->pNext = pNodeTemp;
        m_pTail = pNodeTemp;
    }
}

void CList::RemoveNode(pNode pNodeTemp,CList& CFreeListObj)    
{
    if(pNodeTemp == m_pHead)
    {
        m_pHead = m_pHead->pNext;
    }

    else if(pNodeTemp==m_pTail)
    {
        pNode pNodePre = m_pHead;
        while(pNodePre->pNext!=pNodeTemp)
        {
            pNodePre = pNodePre->pNext;
        }

        pNodePre->pNext = NULL;
        m_pTail = pNodePre;        
    }

    else
    {
        pNode pNodePre = m_pHead;
        while(pNodePre->pNext!=pNodeTemp)
        {
            pNodePre = pNodePre->pNext;
        }

        pNodePre->pNext = pNodeTemp->pNext;    
    }


    m_iNodeCount--;

    if(m_iNodeCount==0)
    {
        m_pTail = NULL;
    }


    if(!CFreeListObj.RecycleNode(pNodeTemp))
    {

        cout<<"Error"<<endl;
    }

}

int CList::GetNodeCount()
{
    if(m_pHead==NULL)
    {
        return 0;
    }

    else
    {
        return m_iNodeCount;
    }
}


bool CList::RecycleNode(pNode pNodeTemp)        //头插法
{
    if(pNodeTemp!=NULL)
    {
        pNodeTemp->pNext = m_pHead;

        m_pHead = pNodeTemp;

        m_iNodeCount++;

        return true;
    }

    return false;

}

pNode CList::AllocateNode()            //从内存池的头部取结点
{
    if(m_pHead == m_pTail)
    {

        m_pTail = NULL;
    }

    pNode pNodeTemp = m_pHead;

    m_pHead = m_pHead->pNext;

    pNodeTemp->pNext = NULL;

    m_iNodeCount--;

    return pNodeTemp;
}

        
bool CList::DestroyList()
{
    if(m_pHead == NULL)
    {
        return false;
    }

    pNode pNodeDel = m_pHead;
    
    while(pNodeDel != NULL)
    {
        m_pHead = pNodeDel->pNext;

        free(pNodeDel);

        pNodeDel = m_pHead;

        m_iNodeCount--;
    }

    return true;
}




int main()
{

    CList CFreeListObj;

    CFile CStudent[4];

    CFile CGraduated;

    CStudent[0].GetFileRoute("Student1.txt");
    CStudent[1].GetFileRoute("Student2.txt");
    CStudent[2].GetFileRoute("Student3.txt");
    CStudent[3].GetFileRoute("Student4.txt");

    CGraduated.GetFileRoute("Graduation.txt");

    int i = 0;

    CStudent[0].FileRead("Student1.txt",CFreeListObj);
    CStudent[1].FileRead("Student2.txt",CFreeListObj);
    CStudent[2].FileRead("Student3.txt",CFreeListObj);
    CStudent[3].FileRead("Student4.txt",CFreeListObj);
    
    i = 0;
    pNode pNodeTemp = NULL;
    bool bOK = true;

    char iMethod = 0;
    int iGrade = 0;
    int iTemp = 0;

    Data DataTemp = {0};


    while(bOK)
    {
        cout<<"┏┅┅┅┅┅┅┅┅┅┅┅┅┅┓"<<endl
            <<"┠       学生管理系统       ┨"<<endl
            <<"┠           V1.0           ┨"<<endl
            <<"┗┅┅┅┅┅┅┅┅┅┅┅┅┅┛"<<endl;
        cout<<"┏┅┅┅┅┅┅┅┅┅┅┅┅┅┓"<<endl
            <<"┠    1.录入学生信息        ┨"<<endl
            <<"┠    2.查看全体学生信息    ┨"<<endl
            <<"┠    3.大四学生毕业处理    ┨"<<endl
            <<"┠    4.查看毕业学生信息    ┨"<<endl
            <<"┠    5.保存学生信息        ┨"<<endl
            <<"┠    6.退出管理系统        ┨"<<endl
            <<"┗┅┅┅┅┅┅┅┅┅┅┅┅┅┛"<<endl;

        cin>>iMethod;


        switch(iMethod)
        {

        case '1':
            cout<<"Input Student's Information:"<<endl;
            cout<<"Grade:";
            cin>>iGrade;
            DataTemp.iGrade = iGrade;
            cout<<"Name:";
            cin>>DataTemp.szName;
            cout<<"Age:";
            cin>>DataTemp.iAge;
            
            
            pNodeTemp = CStudent[iGrade-1].CreateNode(DataTemp,CFreeListObj);

            CStudent[iGrade-1].LinkNode(pNodeTemp);    
            
            switch(iGrade)
            {

            case 1:
                CStudent[iGrade-1].FileWrite("Student1.txt",CFreeListObj);
                break;
            case 2:
                CStudent[iGrade-1].FileWrite("Student2.txt",CFreeListObj);
                break;
            case 3:
                CStudent[iGrade-1].FileWrite("Student3.txt",CFreeListObj);
                break;
            case 4:
                CStudent[iGrade-1].FileWrite("Student4.txt",CFreeListObj);
                break;
            default:
                break;
            }
            

            break;
    

        case '2':
        
                i = 0;
                CStudent[i].InitList();
                CStudent[i].FileRead("Student1.txt",CFreeListObj);
                cout<<"Gread: "<<i+1<<endl;
                cout<<endl
                    <<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;
                CStudent[i++].TravelList();
                cout<<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;

                CStudent[i].InitList();
                CStudent[i].FileRead("Student2.txt",CFreeListObj);
                cout<<"Gread: "<<i+1<<endl;
                cout<<endl
                    <<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;
                CStudent[i++].TravelList();
                cout<<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;

                CStudent[i].InitList();
                CStudent[i].FileRead("Student3.txt",CFreeListObj);
                cout<<"Gread: "<<i+1<<endl;
                cout<<endl
                    <<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;
                CStudent[i++].TravelList();
                cout<<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;

                CStudent[i].InitList();
                CStudent[i].FileRead("Student4.txt",CFreeListObj);
                cout<<"Gread: "<<i+1<<endl;
                cout<<endl
                    <<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;
                CStudent[i++].TravelList();
                cout<<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;
            break;
        case '3':

            CGraduated.FileRead("Graduation.txt",CFreeListObj);

            pNodeTemp = CStudent[3].GetHead();

            while(pNodeTemp!=NULL)
            {
                CGraduated.CreateNode(pNodeTemp->DataTemp,CFreeListObj);
                CGraduated.LinkNode(pNodeTemp);
                pNodeTemp = pNodeTemp->pNext;
            }

            CGraduated.FileWrite("Graduation.txt",CFreeListObj);
            CGraduated.DestroyList();

            
            CStudent[0].FileWrite("Student2.txt",CFreeListObj);
            CStudent[1].FileWrite("Student3.txt",CFreeListObj);
            CStudent[2].FileWrite("Student4.txt",CFreeListObj);
            CStudent[0].InitList();
            CStudent[0].FileWrite("Student1.txt",CFreeListObj);
        
            break;

        case '4':
            cout<<endl
                <<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;
            CGraduated.InitList();
            CGraduated.FileRead("Graduation.txt",CFreeListObj);
            CGraduated.TravelList();
            cout<<"┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅┅"<<endl;
            CGraduated.InitList();
            break;

        case '5':


            CStudent[0].FileWrite("Student1.txt",CFreeListObj);
            CStudent[1].FileWrite("Student2.txt",CFreeListObj);
            CStudent[2].FileWrite("Student3.txt",CFreeListObj);
            CStudent[3].FileWrite("Student4.txt",CFreeListObj);
            
            system("cls");
            break;

        case '6':
            
            CStudent[0].FileWrite("Student1.txt",CFreeListObj);
            CStudent[1].FileWrite("Student2.txt",CFreeListObj);
            CStudent[2].FileWrite("Student3.txt",CFreeListObj);
            CStudent[3].FileWrite("Student4.txt",CFreeListObj);
            
            cout<<"Successful Exit."<<endl;
            bOK = false;
            break;

        default:
            cout<<"Input Error. "<<endl;
            break;
        }


    }

    return 0;
}