用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

*************************************************************

***********************请选择需要进行的操作www.2cto.com******************

*************************************************************

 

w

输入命令不正确！请重新输入！

*************************************************************

***********************请选择需要进行的操作******************

*************************************************************

 

e

请按任意键继续. . .