UVA 814 MTA 模拟

#include<iostream>#include<vector>#include<map>#include<set>#include<string>//#define LOCAL#include<cstdio>using namespace std;void getInformation(string& ss, string user, string & mat){    int k = ss.find('@');    user = ss.substr(0, k);    mat = ss.substr(k+1);}int main(){    #ifdef LOCAL      freopen("in.txt", "r", stdin);      freopen("out.txt", "w", stdout);    #endif // LOCAL    int n;    string s, t, s1, s2, mta1, mta2, user1, user2;    // 对MAT进行整理 ， 变成 姓名@地址的形式.    set<string> adds;    while(cin >> s1 && s1 != "*"){        cin >> s >> n;        while(n--){            cin >> s2;            adds.insert(s2 + "@" + s);        }    }    // 处理发件人的信息， 把姓名和地址分开    while(cin >> s && s != "*")    {    getInformation(s, user1, mta1);  //  关于字符串的处理    vector<string> mta;    map<string, vector<string> > dest;    set<string> vis;    while(cin >> t && t != "*")    {    getInformation(t, user2, mta2);    if(vis.count(t)) continue;  // 重复的收件人， 直接跳过后续工作    vis.insert(t);    if(!dest.count(mta2)) {        mta.push_back(mta2);  dest[mta2] = vector<string>();}        dest[mta2].push_back(t);    }    getline(cin, t);    // 正文的处理，整合成一句话    string data;    while(getline(cin, t) && t != "*") data += "     " + t + "\n";   // 对信息内容的处理    for(int i = 0; i < mta.size(); i++)    {        string mta2 = mta[i];        vector<string> users = dest[mta2];        cout << "Connection between " << mta1 << " and " << mta2 << endl;        cout << "     HELO " << mta1 <<endl << "     250" <<endl;        cout << "     MAIL FROM:<" << s << ">" << endl << "     250" <<endl;        bool ok = false;        for(int i = 0; i < users.size(); i++){            cout << "     RCPT TO:<" << users[i] << ">" <<endl ;            if(adds.count(users[i])) {ok = true; cout << "     250\n";}            else cout << "     550\n";        }        if(ok)        {            cout << "     DATA\n"; cout << "     354\n";            cout << data;            cout << "     .\n" <<"     250\n";        }        cout << "     QUIT\n" << "     221\n";    }    }    return 0;}