1. 程式人生 > 程式設計 >C++容器vector實現通訊錄功能

C++容器vector實現通訊錄功能

之前學習C語言的時候,用連結串列實現過通訊錄的基本功能。最近寫了一個C++版本的通訊錄,參考程式碼如下所示。

main.cpp

/*****************************************************
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;
}

head.h

/*****************************************************
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

book.cpp

/*****************************************************
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

book_ui.cpp

/*****************************************************
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;
}

add_person.cpp

/*****************************************************
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;
 }
}

delete_person.cpp

/*****************************************************
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;
 }
}

change_person.cpp

/*****************************************************
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;
 }
}

find_person.cpp

/*****************************************************
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;
 }
}

display_person.cpp

/*****************************************************
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;
}

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。