用C++容器vector实现通讯录功能

之前学习C语言的时候，用链表实现过通讯录的基本功能。最近写了一个C++版本的通讯录，参考代码如下所示。

main.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： main.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 16时47分52秒Description  ： 主函数Funcion List ： main()*****************************************************/#include "../../include/head.h"personMessage pep;vector<personMessage> person;vector<personMessage>::iterator it;int main(){//personMessage pep;//vector<personMessage> person;char ch = 0;//system("clear");while(ch != 'q'){if((ch != 'a') && (ch != 'c') && (ch != 'd') && (ch != 'f')){system("clear");ch = book_ui();}switch(ch){            case 'a':{ch = add_person();break;}case 'c':{ch = change_person();break;}case 'd':{ch = delete_person();break;}case 'e':{ch = display_person();break;}case 'f':{ch = find_person();break;}case 'q':{cout << "Byebye!" << endl;return 0;break;}default:{cout << "input error!" << endl;break;}}}    return 0;}

head.h

/*****************************************************Copyright (C)： 2017-2018  File name    ： head.hAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 17时11分29秒Description  ： Funcion List ： *****************************************************/#ifndef __HEAD_H__#define __HEAD_H__#include <iostream>#include <string>#include <vector>#include <algorithm>#include <stdio.h>#include <string.h>using namespace std;class personMessage{public:personMessage();personMessage(string s);~personMessage();personMessage& operator=(string s);personMessage& operator=(personMessage& other);/* sort排序算法需要重载'<'，注意加const! */bool operator<(const personMessage& p) const;bool operator>(const personMessage& p) const;bool operator<=(const personMessage& p) const;bool operator>=(const personMessage& p) const;   bool operator==(string s);friend istream& operator>>(istream& in, personMessage& p);friend ostream& operator<<(ostream& out, personMessage& p);int selectFlag;//用来选择哪一个私有成员！private:string name_;string addr_;string phone_;};extern personMessage pep;extern vector<personMessage> person;extern vector<personMessage>::iterator it;extern char book_ui();extern char add_person();extern char change_person();extern char delete_person();extern char display_person();extern char find_person();#endif

book.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： book.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 18时53分19秒Description  ： Funcion List ： *****************************************************/#include "../../include/head.h"personMessage::personMessage() : selectFlag(0){cout << "default coonstructor!" << endl;}personMessage::personMessage(string s){name_ = s;}personMessage::~personMessage(){cout << "destroy person message!" << endl;}#if 1personMessage& personMessage::operator=(string s){name_ = s;return *this;}#endifpersonMessage& personMessage::operator=(personMessage& other){if(this == &other){return *this;}name_  = other.name_;addr_  = other.addr_;phone_ = other.phone_;return *this;}bool personMessage::operator>(const personMessage& p) const{return name_ > p.name_;}bool personMessage::operator>=(const personMessage& p) const{return name_ >= p.name_;}bool personMessage::operator<(const personMessage& p) const{return name_ < p.name_;}bool personMessage::operator<=(const personMessage& p) const{return name_ <= p.name_;}bool personMessage::operator==(string s){if(selectFlag == 1){return name_  == s;}else if(selectFlag == 2){return addr_  == s;}else if(selectFlag == 3){return phone_  == s;}else{return false;}}#if 1istream& operator>>(istream& in, personMessage& p){string name;string addr;string phone;cout << "请输入新的成员名字:" << endl;in >> name;p.name_ = name;cout << "请输入新的成员地址:" << endl;in >> addr;p.addr_ = addr;cout << "请输入新的成员电话:" << endl;in >> phone;p.phone_ = phone;return in;}ostream& operator<<(ostream& out, personMessage& p){out << "名字: " << p.name_ << endl;out << "地址: " << p.addr_ << endl;out << "电话: " << p.phone_ << endl;return out;}#endif

book_ui.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： book_ui.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 16时49分50秒Description  ： Funcion List ： *****************************************************/#include "../../include/head.h"char book_ui(){char ch = 0;cout << " ____________________________________" << endl;cout << "|                                    |" << endl;cout << "|       欢迎进入通讯录系统 v2.0      |" << endl;cout << "|                                    |" << endl;cout << "|====================================|" << endl;cout << "|                                    |" << endl;cout << "|         a. 增加新的成员            |" << endl;cout << "|         c. 修改成员信息            |" << endl;cout << "|         d. 删除成员信息            |" << endl;cout << "|         e. 展示所有成员            |" << endl;cout << "|         f. 查找成员信息            |" << endl;cout << "|         q. 退出通讯录系统          |" << endl;cout << "|____________________________________|" << endl;cout << endl << "请输入你的选择:" << endl;cin >> ch;return ch;}

add_person.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： add_person.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 17时22分56秒Description  ： Funcion List ： *****************************************************/#include "../../include/head.h"char add_person(){cout << "This is add person!" << endl;#if 0getchar();string tmp;getline(cin, tmp);cout << "tmp = " << tmp << endl;pep = tmp;#endif/* 输入新的成员信息 */cin >> pep;cout << pep << endl;/* 向vector插入元素 */person.push_back(pep);cout << "插入成员信息成功!" << endl;char ch = 0;cout << "是否返回主菜单？(y/n)" << endl;getchar();cin >> ch;if(ch == 'y'){return 0;}else if(ch == 'n'){return 'a';}else{cout << "输入错误！" << endl;return 0;}}

delete_person.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： delete_person.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 18时29分33秒Description  ： Funcion List ： *****************************************************/#include "../../include/head.h"char delete_person(){cout << "This is delete person!" << endl;/* 删除成员的信息 */string pep_info;int d_flag = 0;int d_key = 0;cout << "请输入你想要查找的方式(1-姓名/2-地址/3-电话):" << endl;cin >> d_key;switch(d_key){case 1:{cout << "请输入你想要删除成员的名字:" << endl;cin >> pep_info;break;}case 2:{cout << "请输入你想要删除成员的地址:" << endl;cin >> pep_info;break;}case 3:{cout << "请输入你想要删除成员的电话:" << endl;cin >> pep_info;break;}default:{cout << "输入有误！" << endl;return 0;break;}}for(it = person.begin(); it != person.end(); ){it->selectFlag = d_key;if(*it == pep_info){person.erase(person.begin()+d_flag, person.begin()+d_flag+1);cout << "删除成员信息成功!" << endl;}else{++it;d_flag++;}}char ch = 0;cout << "是否返回主菜单?(y/n)" << endl;getchar();cin >> ch;if(ch == 'y'){return 0;}else if(ch == 'n'){return 'd';}else{cout << "输入错误！" << endl;return 0;}}

change_person.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： change_person.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 18时20分15秒Description  ： Funcion List ： *****************************************************/#include "../../include/head.h"char change_person(){cout << "This is change person!" << endl;/* 修改成员的信息 */string pep_info;int ch_flag = 0;int c_key = 0;cout << "请输入你想要查找的方式(1-姓名/2-地址/3-电话):" << endl;cin >> c_key;switch(c_key){case 1:{cout << "请输入你想要修改成员的名字:" << endl;cin >> pep_info;break;}case 2:{cout << "请输入你想要修改成员的地址:" << endl;cin >> pep_info;break;}case 3:{cout << "请输入你想要修改成员的电话:" << endl;cin >> pep_info;break;}default:{cout << "输入有误！" << endl;return 0;break;}}for(it = person.begin(); it != person.end(); ++it){it->selectFlag = c_key;if(*it == pep_info){ch_flag = 1;cin >> *it;cout << "修改成员信息成功!" << endl;}}if(ch_flag != 1){cout << "没有找到该成员!" << endl;}char ch = 0;cout << "是否返回主菜单?(y/n)" << endl;getchar();cin >> ch;if(ch == 'y'){return 0;}else if(ch == 'n'){return 'c';}else{cout << "输入错误！" << endl;return 0;}}

find_person.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： find_person.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 18时21分59秒Description  ： Funcion List ： *****************************************************/#include "../../include/head.h"char find_person(){cout << "This is find person!" << endl;int f_key = 0;int f_flag = 0;/* 输入查找的姓名 */string f_info;cout << "请输入查找方式(1-姓名/2-地址/3-电话)" << endl;cin >> f_key;switch(f_key){case 1:{cout << "请输入你想要查找成员的名字:" << endl;cin >> f_info;break;}case 2:{cout << "请输入你想要查找成员的地址:" << endl;cin >> f_info;break;}case 3:{cout << "请输入你想要查找成员的名字:" << endl;cin >> f_info;break;}default:{cout << "输入有误！" << endl;return 0;break;}}//pep.selectFlag = 2;//it迭代器在变化,不能直接赋值。for(it = person.begin(); it != person.end(); ++it){it->selectFlag = f_key;if(*it == f_info){f_flag = 1;cout << "找到该成员！" << endl;cout << *it << endl;}}if(f_flag != 1){cout << "没有找到该成员!" << endl;}char ch = 0;cout << "是否返回主菜单?(y/n)" << endl;getchar();cin >> ch;if(ch == 'y'){return 0;}else if(ch == 'n'){return 'f';}else{cout << "输入错误！" << endl;return 0;}}

display_person.cpp

/*****************************************************Copyright (C)： 2017-2018  File name    ： display_person.cppAuthor       ： ZhengqijunDate         ： 2017年02月12日 星期日 18时23分04秒Description  ： Funcion List ： *****************************************************/#include "../../include/head.h"char display_person(){cout << "This is display person!" << endl;sort(person.begin(), person.end());for(it = person.begin(); it != person.end(); ++it){cout << *it << endl;}char ch = 0;cout << "按任意键返回" << endl;getchar();cin >> ch;return 0;}