7-15 QQ帐户的申请与登陆 (25分)

问题描述

实现QQ新帐户申请和老帐户登陆的简化版功能。最大挑战是：据说现在的QQ号码已经有10位数了。

输入格式:

输入首先给出一个正整数N（≤10​5​​），随后给出N行指令。每行指令的格式为：“命令符（空格）QQ号码（空格）密码”。其中命令符为“N”（代表New）时表示要新申请一个QQ号，后面是新帐户的号码和密码；命令符为“L”（代表Login）时表示是老帐户登陆，后面是登陆信息。QQ号码为一个不超过10位、但大于1000（据说QQ老总的号码是1001）的整数。密码为不小于6位、不超过16位、且不包含空格的字符串。

输出格式:

针对每条指令，给出相应的信息：

1）若新申请帐户成功，则输出“New: OK”；
2）若新申请的号码已经存在，则输出“ERROR: Exist”；
3）若老帐户登陆成功，则输出“Login: OK”；
4）若老帐户QQ号码不存在，则输出“ERROR: Not Exist”；
5）若老帐户密码错误，则输出“ERROR: Wrong PW”。

输入样例:

5
L 1234567890 myQQ@qq.com
N 1234567890 myQQ@qq.com
N 1234567890 myQQ@qq.com
L 1234567890 myQQ@qq
L 1234567890 myQQ@qq.com

输出样例:

ERROR: Not Exist
New: OK
ERROR: Exist
ERROR: Wrong PW
Login: OK

代码实现

#include<bits/stdc++.h>
using namespace std;

/*****************************************
map的key对应QQ，value对应password。
count()函数用来检测QQ账号是否存在。
find()函数在QQ存在的情况下，用来寻找QQ对应的密码。
******************************************/
int main()
{
    map<string,string> id;
    int n;
    char op;
    string qq, pwd;
    scanf("%d", &n);
    for(int i=0; i<n; ++i) {
        getchar(); //吃掉回车
        scanf("%c", &op);
        if(op == 'N') { //新建
            cin >> qq >> pwd;
            if(id.count(qq)) { //账号已存在
                printf("ERROR: Exist\n");
            }
            else {
                id.insert(pair<string,string>(qq, pwd));
                printf("New: OK\n");
            }

        }
        else { //登录
            cin >> qq >> pwd;
            if(!id.count(qq)) { //账号不存在
                printf("ERROR: Not Exist\n");
            }
            else { //账号存在
                map<string,string>::iterator it;
                it = id.find(qq);
                if(it->second == pwd) //密码正确
                    printf("Login: OK\n");
                else //密码错误
                    printf("ERROR: Wrong PW\n");
            }
        }
    }
    return 0;
}