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

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： main.cpp
Author    ： Zhengqijun
Date     ： 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;
}

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： head.h
Author    ： Zhengqijun
Date     ： 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

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： book.cpp
Author    ： Zhengqijun
Date     ： 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 1
personMessage& personMessage::operator=(string s)
{
 name_ = s;
 return *this;
}
#endif
 
personMessage& 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 1
istream& 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

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： book_ui.cpp
Author    ： Zhengqijun
Date     ： 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;
}

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： add_person.cpp
Author    ： Zhengqijun
Date     ： 2017年02月12日 星期日 17时22分56秒
Description ： 
Funcion List ： 
*****************************************************/
 
#include "../../include/head.h"
 
char add_person()
{
 cout << "This is add person!" << endl;
 
#if 0
 getchar();
 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;
 }
}

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： delete_person.cpp
Author    ： Zhengqijun
Date     ： 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;
 }
}

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： change_person.cpp
Author    ： Zhengqijun
Date     ： 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;
 }
}

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： find_person.cpp
Author    ： Zhengqijun
Date     ： 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;
 }
}

/*****************************************************
Copyright (C)： 2017-2018 
File name  ： display_person.cpp
Author    ： Zhengqijun
Date     ： 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;
}