书本上的一个例题——股票交易系统

/* 1、预处理命令及函数变量 */

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<conio.h>
#include<math.h>
#define MAX 100

typedef struct custstock {            //用户账户数据
    char StockCode[6];                //股票代码
    char StockName[30];                //股票名称
    long StockVal;                    //拥有股票数
}CustStock;

typedef struct custom {                //已注册用户数据
    char CustomerName[20];            //用户名
    char PassWord[6];                //密码
}Customer;

typedef struct stock {                //股票数据
    char StockCode[6];                //股票代码
    char StockName[30];                //股票名称
    long StockVol;                    //总股数
    long StockAva;                    //可交易股数
    long StockNum;                    //股票数
    char chChoice;                    //股票操作选择
}Stock;

int NUM;

void Login(char *,char *,Customer *);
void Register(Customer *);
void Interface_CustOperation(Customer *);
void Interface_StockExchange(Stock *);
void Display_StockMarket(Stock *);
void Menu_Choice(Stock *);
void InitCustData(Customer *);
void Buy(Customer *);
void Sell(Customer *);
void View(Customer *);
void Input_Stock();
void Input_NewStock(Stock *);

/* 2、主函数 */

void main()
{
    char chChoice;
    Stock straSto[MAX],strTemp;
    system("cls");
    printf("


		********************股票交易平台********************

");
    printf("


		需要更新股票信息吗？(y|Y--yes   n|N--no)");
    scanf("%c",&chChoice);
    if('y'==chChoice||'Y'==chChoice)
        Input_Stock();                    //输入股票信息
    Interface_StockExchange(&strTemp);    //股票交易平台
}

/* 3、输入股票信息 */

void Input_Stock()                        //输入股票信息
{
    FILE * fpData;
    int nNum;
    Stock straSto[MAX],strTemp;
    printf("
		输入股票数：");
    scanf("%d",&NUM);
    if((fpData=fopen("Stock_File.dat","wb"))==NULL)
    {
        printf("FILE ERROR
");
        exit(0);
    }
    fwrite(&NUM,sizeof(NUM),1,fpData);
    for(nNum=0;nNum<NUM;nNum++)
    {                                    //输入股票数据
        Input_NewStock(&straSto[nNum]);
        fwrite(&straSto[nNum],sizeof(straSto[nNum]),1,fpData);
    }
    fclose(fpData);
}
void Input_NewStock(Stock * a)            //输入股票数据
{
    system("cls");
    getchar();
    printf("
		请输入股票代码（字符个数<= 6：）");
    gets(a->StockCode);
    printf("
		请输入股票名称：");
    gets(a->StockName);
    printf("
		请输入股票总数：");
    scanf("%ld",&a->StockVol);
    a->StockAva=a->StockVol;            //初始化可交易的股票数
    a->StockNum++;
}

/* 4、股票交易平台主界面 */

void Interface_StockExchange(Stock * a)    //股票交易平台
{
    system("cls");
    printf("


		*************************欢迎进入股票交易平台*******************

");
    printf("			显示股票情况            [v]
");
    printf("			老用户登录            [l]
");
    printf("			新用户注册            [r]
");
    printf("			退出交易系统            [q]
");
    printf("

			请选择：");
    scanf("%c",&a->chChoice);
    if(a->chChoice!='q'&&a->chChoice!='v'&&a->chChoice!='l'&&a->chChoice!='r')
    {
        system("cls");
        Interface_StockExchange(a);        //股票交易平台
    }
    else 
        Menu_Choice(a);                    //执行相关选择
}

/* 5、股票平台操作 */

void Menu_Choice(Stock * a)                //执行相关选择
{
    int nNum;
    FILE * fpCust,*fpData;
    Stock strTemp,straCust[MAX]; 
    Customer strCust;
    char CustomerName[20],PassWord[6];
    switch(a->chChoice)
    {
        case 'v':                        //显示股票情况
            system("cls");
            printf("


	********************股票信息*****************


");
            printf("		 股票名称    股票代码    总股数    可交易的股数

");
            if((fpData=fopen("Stock_File.dat","rb"))==NULL)
            {
                printf("FILE ERROR!
");
                exit(0);
            }
            fread(&NUM,sizeof(NUM),1,fpData);
            for(nNum=0;nNum<NUM;nNum++)
                fread(&straCust[nNum],sizeof(straCust[nNum]),1,fpData);
            fclose(fpData);
            for(nNum=0;nNum<NUM;nNum++)
                Display_StockMarket(&straCust[nNum]);    //显示股票信息
            getch();
            Interface_StockExchange(&strTemp);            //返回股票交易平台
        case 'l':
            system("cls");
            printf("


	********************用户登录*****************


");
            printf("			 用户名：");
            scanf("%s",CustomerName);
            printf("
			 密码：");
            scanf("%s",PassWord);
            getchar();
            Login(CustomerName,PassWord,&strCust);        //用户登录
            Interface_StockExchange(&strTemp);            //返回股票交易平台
        case 'r':
            system("cls");
            printf("


	********************用户注册*****************


");
            Register(&strCust);
            if((fpCust=fopen("customer.dat","ab"))==NULL)
            {
                printf("FILE ERROR!
");
                exit(0);
            }
            fwrite(&strCust,sizeof(strCust),1,fpCust);
            fclose(fpCust);
            InitCustData(&strCust);                        //初始化新用户注册用户库
            Interface_StockExchange(&strTemp);            //返回股票交易平台
        case 'q':
            system("cls");
            printf("





		 谢谢使用股票交易平台

");
            printf("		 再        见

");
            getchar();
            getchar();
            exit(0);
    }
}

/* 6、显示股票信息 */

void Display_StockMarket(Stock * a)            //显示股票信息
{
    printf("
		%-12s%-10s%-10ld%-10ld
",a->StockName,a->StockCode,a->StockVol,a->StockAva);
}

/* 7、用户登录 */

void Login(char * name,char * password,Customer * a)
{
    FILE * fp;
    int Flag;
    if((fp=fopen("customer.dat","rb"))==NULL)
    {
        printf("Read File error!
");
        exit(1);
    }
    while(!feof(fp))                        //查看用户库中的信息
    {
        Flag=fread(a,sizeof(Customer),1,fp);
        if(Flag!=1)
        {
            printf("

		 该用户还未注册！
");
            printf("

		 按任意键返回
");
            getchar();
            break;
        }
        if(strcmp(name,a->CustomerName)==0&&strcmp(password,a->PassWord)==0)
        {
            printf("			 登录成功(按任意键返回)");
            getchar();
            Interface_CustOperation(a);        //登录成功，进入用户操作平台
            break;
        }
    }
    fclose(fp);    
}

/* 8、用户操作平台界面 */

void Interface_CustOperation(Customer * a)    //用户操作平台，完成股票交易
{
    char choice;
    Stock strTemp;
    do
    {
        system("cls");
        printf("


		*************************欢迎进入用户操作平台*******************

");
        printf("

			[b]-------买股票
");
        printf("			[s]-------卖股票
");
        printf("			[v]-------显示用户股票
");
        printf("			[q]-------退出交易
");
        printf("

			请选择：");
        scanf("%c",&choice);
        if(choice!='b'&&choice!='s'&&choice!='v'&&choice!='q')
            break;
        else
        {
            switch(choice)
            {
            case 'b':Buy(a);break;                //买股票
            case 's':Sell(a);break;                //买股票
            case 'v':View(a);break;                //显示用户的股票信息
            case 'q':Interface_StockExchange;    //返回股票交易平台
            }
        }
    } while (1);
}


/* 9、买股票操作 */

void Buy(Customer * a)                                    //买股票
{
    FILE * fpData, *fpCust;
    Stock straShare[MAX];
    CustStock straCuSto[MAX];
    Customer straCust[MAX];
    int nNum;
    char szShareCode[6];
    long nVolume;
    system("cls");
    printf("


	********************股票信息*****************


");
    printf("		 股票名称    股票代码    总股数    可交易的股数

");
    if((fpData=fopen("Stock_File.dat","rb"))==NULL)        //打开股票数据库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    fread(&NUM,sizeof(NUM),1,fpData);
    for(nNum=0;nNum<NUM;nNum++)
        fread(&straShare[nNum],sizeof(straShare[nNum]),1,fpData);
    if((fpCust=fopen(a->CustomerName,"rb"))==NULL)        //打开用户股票账户库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    for(nNum=0;nNum<NUM;nNum++)
        fread(&straCuSto[nNum],sizeof(straCuSto[nNum]),1,fpCust);
    fclose(fpData);
    fclose(fpCust);
    for(nNum=0;nNum<NUM;nNum++)
        Display_StockMarket(&straShare[nNum]);            //显示股票信息
    getchar();
    printf("
		请输入要买入的股票代码：");
    scanf("%s",szShareCode);
    printf("
		请输入股数：");
    scanf("%ld",&nVolume);
    getchar();
    nNum=0;
    while((strcmp(straShare[nNum].StockCode,szShareCode)==0)||nNum<NUM)
    {
        if(strcmp(straShare[nNum].StockCode,szShareCode)==0)
        {
            if(straShare[nNum].StockAva>nVolume)        //符合买股票条件
            {
                straCuSto[nNum].StockVal=straCuSto[nNum].StockVal+nVolume;
                strcpy(straCuSto[nNum].StockName,straShare[nNum].StockName);
                strcpy(straCuSto[nNum].StockCode,straShare[nNum].StockCode);
                straShare[nNum].StockAva=straShare[nNum].StockAva-nVolume;
                if((fpCust=fopen(a->CustomerName,"wb"))==NULL)
                {
                    printf("FILE ERROR!
");
                    exit(0);
                }
                for(nNum=0;nNum<NUM;nNum++)
                    fwrite(&straCuSto[nNum],sizeof(straCuSto[nNum]),1,fpCust);
                if((fpData=fopen("Stock_File.dat","wb"))==NULL)
                {
                    printf("FILE ERROR!
");
                    exit(0);
                }
                fwrite(&NUM,sizeof(NUM),1,fpData);
                for(nNum=0;nNum<NUM;nNum++)
                    fwrite(&straShare[nNum],sizeof(straShare[nNum]),1,fpData);
                fclose(fpData);
                fclose(fpCust);
                break;
            }
            else
            {
                printf("

		该股票可交易份额不足，不能完成本次交易
");
                printf("
		退出本次交易
");
                getchar();
                break;
            }
        }
        else
        {
            nNum++;
            if(nNum==NUM);
            {
                printf("

		输入的股票代码有误……
");
                printf("
		退出本次交易
");
                getchar();
                break;
            }
        }
    }
}

/* 10、卖股票操作 */

void Sell(Customer * a)                                    //卖股票
{
    FILE * fpData, *fpCust;
    Stock straShare[MAX];
    CustStock straCuSto[MAX];
    Customer straCust[MAX];
    int nNum;
    char szShareCode[6];
    long nVolume;
    system("cls");

    if((fpData=fopen("Stock_File.dat","rb"))==NULL)        //打开股票数据库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    fread(&NUM,sizeof(NUM),1,fpData);
    for(nNum=0;nNum<NUM;nNum++)
        fread(&straShare[nNum],sizeof(straShare[nNum]),1,fpData);
    if((fpCust=fopen(a->CustomerName,"rb"))==NULL)        //打开用户股票账户库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    for(nNum=0;nNum<NUM;nNum++)
        fread(&straCuSto[nNum],sizeof(straCuSto[nNum]),1,fpCust);
    fclose(fpData);
    fclose(fpCust);
    printf("


	********************用户股票信息*****************


");
    printf("		 股票名称    股票代码    持股数

");
    for(nNum=0;nNum<NUM;nNum++)
    {
        if(straCuSto[nNum].StockVal!=0)
            printf("		%s		%s	%ld
",straCuSto[nNum].StockName,straCuSto[nNum].StockCode,straCuSto[nNum].StockVal);
    }
    getchar();
    printf("
		请输入要卖的股票代码：");
    scanf("%s",szShareCode);
    printf("
		请输入要卖的股数：");
    scanf("%ld",&nVolume);
    getchar();
    nNum=0;
    while((strcmp(straShare[nNum].StockCode,szShareCode)==0)||nNum<NUM)
    {
        if(strcmp(straShare[nNum].StockCode,szShareCode)==0)
        {
            if(straCuSto[nNum].StockVal>nVolume)        //符合卖股票条件
            {
                straCuSto[nNum].StockVal=straCuSto[nNum].StockVal-nVolume;
                straShare[nNum].StockAva=straShare[nNum].StockAva+nVolume;
                if((fpCust=fopen(a->CustomerName,"wb"))==NULL)
                {
                    printf("FILE ERROR!
");
                    exit(0);
                }
                for(nNum=0;nNum<NUM;nNum++)
                    fwrite(&straCuSto[nNum],sizeof(straCuSto[nNum]),1,fpCust);
                if((fpData=fopen("Stock_File.dat","wb"))==NULL)
                {
                    printf("FILE ERROR!
");
                    exit(0);
                }
                fwrite(&NUM,sizeof(NUM),1,fpData);
                for(nNum=0;nNum<NUM;nNum++)
                    fwrite(&straShare[nNum],sizeof(straShare[nNum]),1,fpData);
                fclose(fpData);
                fclose(fpCust);
                break;
            }
            else
            {
                printf("

		你可交易股票份额不足，不能完成本次交易
");
                printf("
		退出本次交易
");
                getchar();
                break;
            }
        }
        else
        {
            nNum++;
            if(nNum==NUM)
            {
                printf("

		输入的股票代码有误……
");
                printf("
		退出本次交易
");
                getchar();
                break;
            }
        }
    }
}

/* 11、显示用户股票信息 */

void View(Customer * a)                                    //显示用户股票
{
    FILE * fpData, *fpCust;
    CustStock straCuSto[MAX];
    int nNum;
    int flag=1;
    system("cls");

    if((fpData=fopen("Stock_File.dat","rb"))==NULL)        //打开股票数据库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    fread(&NUM,sizeof(NUM),1,fpData);
    if((fpCust=fopen(a->CustomerName,"rb"))==NULL)        //打开用户股票账户库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    for(nNum=0;nNum<NUM;nNum++)
        fread(&straCuSto[nNum],sizeof(straCuSto[nNum]),1,fpCust);
    fclose(fpData);
    fclose(fpCust);
    printf("


	********************用户股票信息*****************


");
    printf("		 股票名称    股票代码    持股数

");
    for(nNum=0;nNum<NUM;nNum++)
    {
        if(straCuSto[nNum].StockVal!=0)
            printf("		%s		%s	%ld
",straCuSto[nNum].StockName,straCuSto[nNum].StockCode,straCuSto[nNum].StockVal);
        flag=0;
    }
    if(flag)
        printf("
		暂无可显示的股票信息");
    getchar();
    getchar();
}

/* 12、新用户注册 */

void Register(Customer * a)                        //新用户注册
{
    system("cls");
    getchar();
    printf("

	********************用户注册*****************

");
    printf("
			请输入用户名：");
    scanf("%s",a->CustomerName);
    printf("
			请输入密码：");
    scanf("%s",a->PassWord);
    getchar();
    system("cls");
    printf("

你已注册成功
");
    getchar();;
}

void InitCustData(Customer * a)                            //初始化新注册用户库
{
    int nNum;
    CustStock straCuSto[MAX];
    FILE * fpCust, * fpData;
    if((fpData=fopen("Stock_File.dat","rb"))==NULL)        //打开股票数据库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    fread(&NUM,sizeof(NUM),1,fpData);
    if((fpCust=fopen(a->CustomerName,"wb"))==NULL)        //打开用户股票账户库
    {
        printf("FILE ERROR!
");
        exit(0);
    }
    for(nNum=0;nNum<NUM;nNum++)
    {
        strcpy(straCuSto[nNum].StockName,"");
        strcpy(straCuSto[nNum].StockCode,"");
        straCuSto[nNum].StockVal=0;
        fwrite(&straCuSto[nNum],sizeof(straCuSto[nNum]),1,fpCust);
    }
    fclose(fpCust);
    fclose(fpData);
}