The Letter Carrier's Rounds（摘）

#include<iostream>
#include<map>
#include<string>
#include<vector>
#include<set>
using namespace std;

void parse_address(const string&s,string& user,string&mta)
{
    int k=s.find(‘@‘);
    user=s.substr(0,k);            //将user1@mta1 分解开来
    mta=s.substr(k+1);
}
int main()
{
    int n;
    string s,t,user1,mta1,user2,mta2;
    set<string>address;
    
    while(cin>>s&&s!="*"){                //mta下的用户信息
        cin>>s>>n;
        while(n--){
            cin>>t;
            address.insert(t+‘@‘+s);
        }
    }
    while(cin>>s&&s!="*"){
        parse_address(s,user1,mta1);         //发件人

        vector<string>mta;                   //所有需要连接的mta，按照输入顺序
        map<string,vector<string>>dest;      //每个mta需要发送的用户
        set<string>vis;

        while(cin>>t&&t!="*"){                  //收件人
            parse_address(t,user2,mta2);
            if(vis.count(t))continue;          //收件人重复则不插入
            vis.insert(t);
            if(!dest.count(mta2)){             //mta向量中没有该mta就插入并将向量与map中的string mta对应起来
                mta.push_back(mta2);
                dest[mta2]=vector<string>();
            }
            dest[mta2].push_back(t);        //插入user@mta 完整形式
        }
        getline(cin,t);               //吃掉"*"

        string data;
        while(getline(cin,t)&&t[0]!=‘*‘)data+="  "+t+"\n";      //将整个邮件存入一个字符串data中

        //cout<<"***"<<data<<"***"<<endl;
        for(int i=0;i<mta.size();i++){
            string mta2=mta[i];
            vector<string>users=dest[mta2];                 //users即为dest[mata2]，mata2对应的需要发送的用户
            cout<<"Connection between "<<mta1<<" and "<<mta2<<endl;
            cout<<"  HELO "<<mta1<<"\n";
            cout<<"  250\n";
            cout<<"  MALL FROM:<" <<s<<">\n";
            cout<<"  250\n";
            bool ok=false;
            for(int i=0;i<users.size();i++){
                cout<<"RCPT TO:<"<<users[i]<<">\n";
                if(address.count(users[i])){
                    ok=true;
                    cout<<"  250\n";
                }
                else cout<<"  500\n";
            }
            if(ok){
                cout<<"  DATA\n";
                cout<<"  254\n";
                cout<<data;
                cout<<".\n";
                cout<<"  250\n";
            }
            cout<<"  QUIT\n";
            cout<<"  221\n";
        }
    }
    //system("pause");
    return 0;
}