c++课程设计，职工管理系统

#include <iostream>
#include <string>
#include <set>
#include <windows.h>
#include <cstdio>
#include <conio.h>
#include <cmath>
#include <fstream>
#define I 12 //爱心的上半部分两个小半圆的半径
#define R 240 //爱心的下半部分圆弧曲线半径
using namespace std;
class zhigong//职工类，表示单个职工
{
string nam;//姓名
string gh;//工号
bool sex;//性别
string sr;//生日
string xl;//学历
string zw;//职务
int gz;//工资
string zhu;//住址
string tel;//电话
public:
bool operator < (const  zhigong &b) const;//为set提供排序方式
zhigong(string _nam,string _gh,bool _sex,string _sr,string _xl,string _zw,int _gz,string _zhu,string _tel);
zhigong();
zhigong(string x);//为工号查找提供简易构造函数
void prin();//屏幕输出
friend ostream& operator << (ostream &out,const zhigong& b);//为文件对象提供输出
void change();//修改
string na();//为姓名模糊查找提供nam
};
class zgxt:public set<zhigong>//继承STL_set作为程序主题类表示职工系统类，作为平衡红黑二叉树树存储
{
public:
void sdin();//手动插入
void shuc();//输出
zgxt::iterator mhfind(string _na);//模糊查找姓名
} acm;//定义一个对象acm表示职工的集合
void gotoxy(int x,int y)
{

COORD loc;
loc.X  =  y;
loc.Y  =  x;
SetConsoleCursorPosition(GetStdHandle(STD_OUTPUT_HANDLE), loc);
}//坐标定位函数
void xinxing()//欢迎界面
{
int i,j,e,a;
for(i=1,a=I; i<I/2; i++,a--)//爱心的上半部分 为两个实心半圆
{
for(j=(int) ( I-sqrt((double)(I*I-(a-i)*(a-i))) ); j>0; j--)
printf(" ");
for(e=1; e<=2*sqrt((double)(I*I-(a-i)*(a-i))); e++)
printf("\3");
for(j=(int) ( 2*( I-sqrt((double)(I*I-(a-i)*(a-i))) ) ); j>0; j--)
printf(" ");
for(e=1; e<=2*sqrt( (double) (I*I-(a-i)*(a-i)) ); e++)
printf("\3");
printf("\n");
}
for(i=1; i<56; i++)//爱心的中间一行 可以打印一些标记
{
if(i==14)
{
printf("\3\3\3\3\3 ~Welcome~ \3\3\3\3\3\3");
i+=30;
}
printf("\3");
}
printf("\n");
for(i=1; i<=R/2; i++)//爱心的下半部分 以R为半径的左右两条圆弧曲线 R设置为比较大
{
if(i%2||i%3)continue;//因为R很大 两条曲线应该删去中间的一些点 这样就会更像爱心
for(j=(int) ( R-sqrt( (double) (R*R-i*i) ) ); j>0; j--)
printf(" ");
for(e=1; e<=2*( sqrt( (double)(R*R-i*i) ) - (R-2*I) ); e++)
printf("\3");
printf("\n");
}
gotoxy(0,79);//以下是控制变色
system("color a");
if(kbhit())
return;
Sleep(500);
system("color b");
if(kbhit())
return;
Sleep(500);
system("color c");
if(kbhit())
return;
Sleep(500);
system("color d");
system("color e");
if(kbhit())
return;
Sleep(500);
system("color f");
if(kbhit())
return;
Sleep(500);
system("color 0");
if(kbhit())
return;
Sleep(500);
system("color 1");
if(kbhit())
return;
Sleep(500);
system("color 2");
system("color 3");
if(kbhit())
return;
Sleep(500);
system("color 4");
if(kbhit())
return;
Sleep(500);
system("color 5");
if(kbhit())
return;
Sleep(500);
system("color 6");
if(kbhit())
return;
Sleep(500);
system("color 7");
system("color 8");
if(kbhit())
return;
Sleep(500);
system("color 9");
system("color ab");
if(kbhit())
return;
Sleep(500);
system("color ac");
if(kbhit())
return;
Sleep(500);
system("color ad");
if(kbhit())
return;
Sleep(500);
system("color ae");
if(kbhit())
return;
Sleep(500);
system("color 1f");
system("cls");
}
void caidan()//显示菜单
{
system("cls");
gotoxy(2,28);
puts("职工信息管理系统");
puts("");
puts("                                 1.插入");
puts("                                 2.删除");
puts("                                 3.显示");
puts("                                 4.修改");
puts("                                 5.查询");
puts("                                 6.结束");
}
int main()
{
acm.clear();
fstream in("1.txt",ios::in);//打开文件
if(!in)//如果1.txt不存在
{
in.open("1.txt",ios::out);//就用out建立一个
in.close();
in.open("1.txt",ios::in);//让输入流等于1.txt
}
string nam_;//姓名
string gh_;//工号
bool sex_;//性别
string sr_;//生日
string xl_;//学历
string zw_;//职务
int gz_;//工资
string zhu_;//住址
string tel_;//电话
while(in>>nam_)
{
in>>gh_;
int ac;
in>>ac;
sex_=ac;
in>> sr_>>xl_ >>zw_ >>gz_ >>zhu_ >>tel_;
zhigong qwe(nam_,gh_,sex_,sr_,xl_,zw_,gz_,zhu_,tel_);//文件读入
acm.insert(qwe);//存入集合中
}
in.close();//关闭文件
xinxing();//心形界面
system("color 1e");
caidan();
string k;
while(cin>>k)
{
if(k.size()==1)
{
switch(k[0])
{
case '1':
acm.sdin();//手动插入
break;
case '2':
{
cout<<"请输入要删除的工号,输入out退出"<<endl;
string _gh;
while(cin>>_gh)
{
if(_gh=="out")
break;
zhigong x(_gh);
zgxt::iterator p=acm.find(x);//set提供了一个find()返回的是符合条件的迭代器
if(p==acm.end())
{
cout<<"输入有误请重新输入"<<endl;
Sleep(500);
cout<<"请输入要删除的工号"<<endl;
continue;
}
zhigong ak=*p;
ak.prin();
string w;
cout<<"该职工是否是您需要的？（y/n）：";
bool adc=true;
while(cin>>w)
{
if(w=="Y"||w=="y")
{
acm.erase(*p);
cout<<"删除成功"<<endl;
system("pause");
break;
}
if(w=="N"||w=="n")
{
cout<<"那么你想干啥？\n1：重新输入\n2：退出"<<endl;
int apc;
while(cin>>apc)
{
if(apc==1)
{
adc=false;
break;
}
if(apc==2)
break;
cout<<"输入有误"<<endl;
}
break;
}
cout<<"输入有误"<<endl;
}
if(!adc)
{
cout<<"请输入要删除的工号,输入out退出"<<endl;
continue;
}
else
break;
}
break;
}
case '3':
acm.shuc();//输出集合
break;
case '4':
{
cout<<"请输入要修改的工号,输入out退出"<<endl;
string _gh;
while(cin>>_gh)
{
if(_gh=="out")
break;
zhigong x(_gh);
zgxt::iterator p=acm.find(x);
if(p==acm.end())
{
cout<<"输入有误请重新输入"<<endl;
Sleep(500);
cout<<"请输入要删除的工号"<<endl;
continue;
}
x=*p;
acm.erase(*p);
x.change();
acm.insert(x);
break;
}
break;
}
case '5':
{
zgxt::iterator p;
cout<<"请输入姓名关键字"<<endl;//姓名查找
string s;
cin>>s;
p=acm.mhfind(s);
if(p==acm.end())
{
cout<<"不存在"<<endl;
system("pause");
}
else
{
cout<<"您想干什么？"<<endl;
cout<<"1: 删除\n2: 修改\n3: 只是看一下"<<endl;
int ss;
while(cin>>ss)
{
if(ss==1)
{
zhigong x=*p;
acm.erase(x);
cout<<"删除成功"<<endl;
system("pause");
break;
}
if(ss==2)
{
zhigong x=*p;
acm.erase(x);
x.change();
acm.insert(x);
break;
}
if(ss==3)
break;
cout<<"输入有误请重新输入"<<endl;
}
}
break;
}
case '6':
{
fstream out("1.txt",ios::out);//打开文件
zgxt::iterator p=acm.begin();
while(p!=acm.end())//把zgxt里面的所有元素都存入文件中
{
zhigong lx=*p;
out<<lx;
p++;
}
out.close();//关闭文件
return 0;
}
default :
{
cout<<"输入有误，请重新输入"<<endl;
Sleep(500);
break;
}
}
caidan();
continue;
}
cout<<"输入有误，请重新输入"<<endl;
Sleep(500);
caidan();
}
return 0;
}
void zhigong::change()//修改
{
prin();
cout<<"您需要修改几处？";
int n;
cin>>n;
while(n--)
{
cout<<"您需要修改的是:"<<"\n"
<<"1：姓名"<<"\n"
<<"2：工号"<<"\n"
<<"3：性别"<<"\n"
<<"4：生日"<<"\n"
<<"5：学历"<<"\n"
<<"6：职务"<<"\n"
<<"7：工资"<<"\n"
<<"8：住址"<<"\n"
<<"9：电话"<<endl;
string k;
while(cin>>k)
{
if(k.size()==1&&k[0]>='1'&&k[0]<='9')
break;
cout<<"输入有误请重新输入"<<endl;
}
switch(k[0]-'0')
{
case 1:
{
cout<<"姓名： ";
cin>>nam;
break;
}
case 2:
{
cout<<"工号： ";
string _gh;
while(cin>>_gh)
{
if(gh==_gh)
break;
zhigong x(_gh);
if(!acm.count(x))//判断当前工号是否存在
break;
cout<<"该工号重复,请重新输入"<<endl;
cout<<"工号： ";
}
gh=_gh;
break;
}
case 3:
{
cout<<"请选择性别：(男：1/女 0) : ";
string kt;
while(cin>>kt)
{
if(kt=="0")
{
sex=false;
break;
}
if(kt=="1")
{
sex=true;
break;
}
cout<<"输入有误,请重新输入"<<endl;
cout<<"请选择性别：(男：1/女 0) : ";
}
break;
}
case 4:
{
cout<<"生日： ";
cin>>sr;
break;
}
case 5:
{
cout<<"学历： ";
cin>>xl;
break;
}
case 6:
{
cout<<"职务： ";
cin>>zw;
break;
}
case 7:
{
cout<<"工资： ";
cin>>gz;
break;
}
case 8:
{
cout<<"住址： ";
cin>>zhu;
break;
}
case 9:
{
cout<<"电话： ";
cin>>tel;
break;
}
}
}
cout<<"修改成功"<<endl;
system("pause");
}
void zgxt::sdin()
{
int n,cont=0;
cout<<"请输入要插入的职工个数 :";
cin>>n;
while(n--)
{
cout<<"第"<<++cont<<"个职工"<<endl;
cout<<"姓名： ";
string _nam;
cin>>_nam;
cout<<"工号： ";
string _gh;
while(cin>>_gh)
{
zhigong x(_gh);
if(!this->count(x))//判重
break;
cout<<"该工号重复,请重新输入"<<endl;
cout<<"工号： ";
}
cout<<"请选择性别：(男：1/女 0) : ";
string k;
bool _sex=false;
while(cin>>k)
{
if(k=="0")
{
_sex=false;
break;
}
if(k=="1")
{
_sex=true;
break;
}
cout<<"输入有误,请重新输入"<<endl;
cout<<"请选择性别：(男：1/女 0) : ";
}
cout<<"生日： ";
string _sr;
cin>>_sr;
cout<<"学历： ";
string _xl;
cin>>_xl;
cout<<"职务： ";
string _zw;
cin>>_zw;
cout<<"工资： ";
int _gz;
cin>>_gz;
cout<<"住址： ";
string _zhu;
cin>>_zhu;
cout<<"电话： ";
string _tel;
cin>>_tel;
cout<<endl;
zhigong x(_nam,_gh,_sex,_sr,_xl,_zw,_gz,_zhu,_tel);
this->insert(x);//插入
}
}
bool zhigong::operator < (const  zhigong &b)const//为set提供排序方式
{
return gh<b.gh;
}
zhigong::zhigong(string _nam,string _gh,bool _sex,string _sr,string _xl,string _zw,int _gz,string _zhu,string _tel)
:nam(_nam),gh(_gh),sex(_sex),sr(_sr),xl(_xl),zw(_zw),gz(_gz),zhu(_zhu),tel(_tel) {}//构造函数
zhigong::zhigong() {}
zhigong::zhigong(string x)//为工号查找提供简易构造函数
{
nam="",gh=x,sex=false,sr="",xl="",zw="",gz=0,zhu="",tel="";
}
void zhigong::prin()//屏幕输出
{
cout<<"姓名： "<<nam<<endl;
cout<<"工号： "<<gh<<endl;
cout<<"性别： ";
if(sex)
cout<<"男"<<endl;
else
cout<<"女"<<endl;
cout<<"生日： "<<sr<<endl;
cout<<"学历： "<<xl<<endl;
cout<<"职务： "<<zw<<endl;
cout<<"工资： "<<gz<<endl;
cout<<"住址： "<<zhu<<endl;
cout<<"电话： "<<tel<<"\n"<<endl;
}
ostream& operator << (ostream &out,const zhigong& b)//为文件对象提供输出
{
out<<b.nam<<endl<<b.gh<<endl<<b.sex<<endl<<b.sr<<endl<<b.xl<<endl<<b.zw<<endl<<b.gz<<endl<<b.zhu<<endl<<b.tel<<endl;
return out;
}
string zhigong::na()//为姓名模糊查找提供nam
{
return nam;
}
void zgxt::shuc()//输出
{
system("cls");
zgxt::iterator p = this->begin();//迭代器
cout<<"共"<<this->size()<<"个成员"<<endl;
while(p!=this->end())
{
zhigong x;
x=*p;
p++;
x.prin();
}
system("pause");
}
zgxt::iterator zgxt::mhfind(string _na)//模糊查找姓名
{
zgxt::iterator p=this->begin();
while(p!=this->end())
{
zhigong x=*p;
if(x.na().find(_na)!=0xffffffff)//string 提供了find()函数，如果未找到就返回一个0xffffffff
{
cout<<"找到目标"<<endl;
x.prin();
cout<<"该目标是否是您需要的？(y/n) ";
string s;
while(cin>>s)
{
if(s=="y"||s=="Y")
return p;
if(s=="N"||s=="n")
break;
cout<<"输入有误"<<endl;
}
}
p++;
}
return this->end();
}