用vector、 multimap、 list容器实现好友列表的各种操作 C++

用vector、 multimap、 list容器实现好友列表的各种操作 C++

--------------------------------------------------BuddyList.h------------------------------------------------#include<map>
#include<string>
#include<list>

//using std::multimap;
//using std::string;
//using std::list;

using namespace std;

struct Info
{
 string id;
 string name;
 string otherInfo;
};

class BuddyList
{
public:
 BuddyList();

 Info getMyInfo();

 void setMyInfo(const Info &info);

 void addBuddy(const Info &friInfo);

 bool remove(const string &friId);

 bool isBuddy(const string &friId);

 list<Info> getBuddies() const;

protected:
 multimap<string,Info>mBuddies;

private:
 Info myInfo;

 //BuddyList(const BuddyList &src);

 //BuddyList & operator=(const BuddyList &rhs);
};

BuddyList::BuddyList(){};

Info BuddyList::getMyInfo()
{
 return myInfo;
}

void BuddyList::setMyInfo(const Info &info)
{
 myInfo = info;
}

void BuddyList::addBuddy(const Info &friInfo)
{
 if( !isBuddy(friInfo.id))
  mBuddies.insert(make_pair(myInfo.id,friInfo));
}

bool BuddyList::remove(const std::string &friId)
{
 if(!isBuddy(friId))
  return false;
 multimap<string,Info>::iterator start,end;
 start = mBuddies.lower_bound(myInfo.id);
 end = mBuddies.upper_bound(myInfo.id);

 for(;start != end; ++start)
 {
  if(start->second.id == friId)
  {
   mBuddies.erase(start);
   return true;
  }
 }
}

bool BuddyList::isBuddy(const std::string &friId)
{
 multimap<string,Info>::iterator start,end;
 start = mBuddies.lower_bound(myInfo.id);
 end = mBuddies.upper_bound(myInfo.id);

 for(;start != end; ++start)
 {
  if(start->second.id == friId)
   return true;
 }
 return false;
}

list<Info> BuddyList::getBuddies() const
{
 /*
 pair<multimap<string,Info>::const_iterator,
  multimap<string,Info>::const_iterator> its;

 its = mBuddies.equal_range(myInfo.id);
   */
  
 list<Info> buddies;

 multimap<string,Info>::iterator it;

 for(multimap<string,Info>::const_iterator it = mBuddies.begin(); it != mBuddies.end(); ++it)
  buddies.push_back(it->second);
 return buddies;
}

 

-------------------------------------------------主函数.cpp----------------------------------------------------// 好友列表实现.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include<iostream>
#include "BuddyList.h"
#include<string>
#include<cstring>
#include<vector>
using namespace std;

int _tmain(int argc, _TCHAR* argv[])
{
 Info info1,info2;
 vector<BuddyList> usersBudList;
 cout<<"***********************请选择需要进行的操作******************"<<endl;
 cout<<"---------------建立所有用户好友列表(C/c)------------------"<<endl;
 cout<<"-------------------新增用户好友(I/i)----------------------"<<endl;
 cout<<"---------------查找指定用户所有好友(S/s)------------------"<<endl;
 cout<<"---------------删除指定用户某位好友(R/r)------------------"<<endl;
 cout<<"------------------------退出(E/e)-------------------------"<<endl;
 cout<<endl<<endl<<endl;
 char cmd;
 cin>>cmd;
 while(1)
 {
  if(cmd=='C'||cmd=='c')
  {
   cout<<"%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%"<<endl;
   cout<<"请输入所有用户"<<endl;
   cout<<"%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%"<<endl;

   while(1)
   {
    cout<<endl<<endl;
    cout<<"&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&"<<endl;
    cout<<"请输入账号（以0结束）：";
    cin>>info1.id;
    if(info1.id=="0")
     break;
    cout<<"请输入他的姓名：";
    cin>>info1.name;
    cout<<"请输入他的其他的信息：";
    cin>>info1.otherInfo;
    BuddyList buddyList;
    buddyList.setMyInfo(info1);
    ///usersBudList.push_back(buddyList);
    cout<<endl;
    cout<<"&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&"<<endl;
    cout<<"请输入他所有好友的信息"<<endl;
    while(1)
    {
     cout<<"**********************************"<<endl;
     cout<<"请输入好友账号（以0结束）：";
     cin>>info2.id;
     if(info2.id=="0")
      break;
     cout<<"请输入他的姓名：";
     cin>>info2.name;
     cout<<"请输入他的其他的信息：";
     cin>>info2.otherInfo;
     buddyList.addBuddy(info2);
    }
    usersBudList.push_back(buddyList);
   }
  }
  else if(cmd == 'I' || cmd=='i')
  {
   cout<<endl<<endl;
   string userId;
   cout<<"****************************************"<<endl;
   cout<<"请输入用户账号（以0结束）："<<endl;
   cin>>userId;
   if(userId=="0")
    break;
   vector<BuddyList>::iterator it1;
   int i =0;
   for( it1= usersBudList.begin(); it1 != usersBudList.end(); ++it1,++i)
   {
    if(it1->getMyInfo().id == userId)
    {
     cout<<endl;
     cout<<"**********************************"<<endl;
     cout<<"请输入新增好友的信息"<<endl;
     Info info1;
     while(1)
     {
      cout<<"**********************************"<<endl;
      cout<<"请输入好友账号（以0结束）：";
      cin>>info1.id;
      if(info1.id=="0")
       break;
      cout<<"请输入他的姓名：";
      cin>>info1.name;
      cout<<"请输入他的其他的信息：";
      cin>>info1.otherInfo;
      BuddyList *pBuddyList = &usersBudList.at(i);
      //cout<<pBuddyList->getMyInfo().id<<endl;
      if(pBuddyList->isBuddy(info1.id))
      {
       cout<<"他的好友列表中已存在此账号！"<<endl;
       continue;
      }
      pBuddyList->addBuddy(info1);
      cout<<"添加成功！"<<endl;
     }
     break;
    }
   }
   if(it1 == usersBudList.end())
     cout<<"无此账号的用户！"<<endl;
  }
  else if(cmd=='S'||cmd=='s')
  {
   cout<<endl<<endl;
   string userId;
   list<Info> userList;
   while(1)
   {
    cout<<"****************************************"<<endl;
    cout<<"请输入用户账号（以0结束）："<<endl;
    cin>>userId;
    if(userId=="0")
     break;
    vector<BuddyList>::iterator it1;
    for( it1= usersBudList.begin(); it1 != usersBudList.end(); ++it1)
    {
     if(it1->getMyInfo().id == userId)
     {

      userList = it1->getBuddies();
      if(userList.size()==0)
      {
       cout<<"此账号用户的好友列表为空！"<<endl;
       break;
      }
      cout<<"他的好友有："<<endl;
      for(list<Info>::iterator it2 = userList.begin(); it2 != userList.end(); ++it2)
      {
       cout<<"账号："<<it2->id<<"  姓名："<<it2->name<<"  其他信息："<<it2->otherInfo<<endl;
      }
      break;
     }
    }
    if(it1 == usersBudList.end())
     cout<<"无此账号的用户！"<<endl;
   }
  }
  else if(cmd=='R'||cmd=='r')
  {
   cout<<endl<<endl;
   string userId;
   string friId;
   list<Info> userList;
   while(1)
   {
    cout<<"****************************************"<<endl;
    cout<<"请输入用户账号（以0结束）："<<endl;
    cin>>userId;
    if(userId=="0")
     break;
    vector<BuddyList>::iterator it1 = usersBudList.begin();
    for(; it1 != usersBudList.end(); ++it1)
    {
     if(it1->getMyInfo().id == userId)
     {
      cout<<endl;
      cout<<"****************************************"<<endl;
      cout<<"请输入需要删除的好友的账号："<<endl;
      cin>>friId;
      bool flag = it1->remove(friId);
      if(flag)
       cout<<"删除成功！"<<endl;
      else
       cout<<"他的好友中没有此账号！"<<endl;
      break;
     }
    }
    if(it1 == usersBudList.end())
     cout<<"无此账号的用户！"<<endl;
   }
  }
  else if(cmd=='E'||cmd=='e')
  {
   break;
  }
  else
  {
   cout<<"输入命令不正确！请重新输入！"<<endl;
  }
  cout<<"*************************************************************"<<endl;
  cout<<"***********************请选择需要进行的操作******************"<<endl;
  cout<<"*************************************************************"<<endl;
  cout<<endl<<endl<<endl;
  cin>>cmd;
 }
 system("pause");
 return 0;
}

 

----------------------------------------------程序测试---------------------------------------------------------

***********************请选择需要进行的操作******************
---------------建立所有用户好友列表(C/c)------------------
-------------------新增用户好友(I/i)----------------------
---------------查找指定用户所有好友(S/s)------------------
---------------删除指定用户某位好友(R/r)------------------
------------------------退出(E/e)-------------------------

 

c
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
请输入所有用户
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
请输入账号（以0结束）：2008550820
请输入他的姓名：heyong
请输入他的其他的信息：cs

&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
请输入他所有好友的信息
**********************************
请输入好友账号（以0结束）：0894042210
请输入他的姓名：luoyao
请输入他的其他的信息：ac
**********************************
请输入好友账号（以0结束）：2008550802
请输入他的姓名：zhouqian
请输入他的其他的信息：cs
**********************************
请输入好友账号（以0结束）：2008552980
请输入他的姓名：lina
请输入他的其他的信息：sw
**********************************
请输入好友账号（以0结束）：0

&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
请输入账号（以0结束）：0894042210
请输入他的姓名：luoyao
请输入他的其他的信息：ac

&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
请输入他所有好友的信息
**********************************
请输入好友账号（以0结束）：0897508652
请输入他的姓名：liuhuan

请输入他的其他的信息：tg
**********************************
请输入好友账号（以0结束）：0812345678

请输入他的姓名：lijun
请输入他的其他的信息：tf
**********************************
请输入好友账号（以0结束）：0

&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&
请输入账号（以0结束）：0
*************************************************************
***********************请选择需要进行的操作******************
*************************************************************

 

i

****************************************
请输入用户账号（以0结束）：
0894042210

**********************************
请输入新增好友的信息
**********************************
请输入好友账号（以0结束）：2008550819
请输入他的姓名：lixun
请输入他的其他的信息：cs
0894042210
添加成功！
**********************************
请输入好友账号（以0结束）：2008550820
请输入他的姓名：heyong
请输入他的其他的信息：cs
0894042210
添加成功！
**********************************
请输入好友账号（以0结束）：2008550820
请输入他的姓名：heyong
请输入他的其他的信息：cs
0894042210
他的好友列表中已存在此账号！
**********************************
请输入好友账号（以0结束）：0
*************************************************************
***********************请选择需要进行的操作******************
*************************************************************

 

s

****************************************
请输入用户账号（以0结束）：
2008550820
他的好友有：
账号：0894042210  姓名：luoyao  其他信息：ac
账号：2008550802  姓名：zhouqian  其他信息：cs
账号：2008552980  姓名：lina  其他信息：sw
****************************************
请输入用户账号（以0结束）：
0894042210
他的好友有：
账号：0897508652  姓名：liuhuan  其他信息：tg
账号：08123456  姓名：lijun  其他信息：tf
账号：2008550819  姓名：lixun  其他信息：cs
账号：2008550820  姓名：heyong  其他信息：cs
****************************************
请输入用户账号（以0结束）：
2008550819
无此账号的用户！
****************************************
请输入用户账号（以0结束）：
0
*************************************************************
***********************请选择需要进行的操作******************
*************************************************************

 

r

****************************************
请输入用户账号（以0结束）：
0894042210

****************************************
请输入需要删除的好友的账号：
08123456

删除成功！
****************************************
请输入用户账号（以0结束）：
0
*************************************************************
***********************请选择需要进行的操作******************
*************************************************************

 

s

****************************************
请输入用户账号（以0结束）：
0894042210
他的好友有：
账号：0897508652  姓名：liuhuan  其他信息：tg
账号：2008550819  姓名：lixun  其他信息：cs
账号：2008550820  姓名：heyong  其他信息：cs
****************************************
请输入用户账号（以0结束）：
0
*************************************************************
***********************请选择需要进行的操作******************
*************************************************************

 

w
输入命令不正确！请重新输入！
*************************************************************
***********************请选择需要进行的操作******************
*************************************************************

 

e
请按任意键继续. . .