QQ帐户的申请与登陆

QQ帐户的申请与登陆

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

输入格式:

输入首先给出一个正整数N（≤ 10⁵），随后给出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

 

解题思路

　　先给出std::unordered_map实现的AC代码：

#include <iostream>
#include <vector>
#include <string>
#include <unordered_map>
#include <algorithm>
using namespace std;

int main() {
    int n;
    cin >> n;
    unordered_map<string, string> mp;
    
    for (int i = 0; i < n; i++) {
        string op, id, pw;
        cin >> op >> id >> pw;
        if (op == "N") {
            if (mp.count(id) == 1) {
                cout << "ERROR: Exist\n";
            }
            else {
                mp[id] = pw;
                cout << "New: OK\n";
            }
        }
        else if (op == "L") {
            if (mp.count(id) == 0) {
                cout << "ERROR: Not Exist\n";
            }
            else if (mp[id] == pw) {
                cout << "Login: OK\n";
            }
            else {
                cout << "ERROR: Wrong PW\n";
            }
        }
    }
    
    return 0;
}

　　然后是手写的哈希散列AC代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>

const int MAXN = 17;

struct Data {
    char id[MAXN], pw[MAXN];
};

struct HashTable {
    int tableSize;
    std::vector<std::vector<Data> > table;
};

HashTable *createTable();
int prime();
bool find(Data &data, HashTable *ht);
void insert(Data &data, HashTable *ht);

int main() {
    int n;
    scanf("%d", &n);
    HashTable *ht = createTable();
    
    for (int i = 0; i < n; i++) {
        char op;
        scanf("\n%c", &op);
        Data data;
        scanf(" %s %s", data.id, data.pw);
        
        if (op == 'N') {
            if (!find(data, ht)) {
                insert(data, ht);
                puts("New: OK");
            }
            else {
                puts("ERROR: Exist");
            }
        }
        else if (op == 'L') {
            if (!find(data, ht)) {
                puts("ERROR: Not Exist");
            }
            else {
                int pos = atoi(data.id) % 10000;
                std::vector<Data>::iterator it = ht->table[pos].begin();
                for ( ; strcmp(it->id, data.id); it++);
                if (strcmp(it->pw, data.pw) == 0) puts("Login: OK");
                else puts("ERROR: Wrong PW");
            }
        }
    }
    
    return 0;
}

HashTable *createTable() {
    HashTable *ht = new HashTable;
    ht->tableSize = prime();
    ht->table.resize(ht->tableSize);
    
    return ht;
}

int prime() {
    int i = 10001;
    while (true) {
        int j = (int)sqrt(10000);
        for ( ; j > 2; j--) {
            if (i % j == 0) break;
        }
        if (j == 2) break;
        i += 2;
    }
    
    return i;
}

bool find(Data &data, HashTable *ht) {
    int pos = atoi(data.id) % 10000;
    for (std::vector<Data>::iterator it = ht->table[pos].begin(); it != ht->table[pos].end(); it++) {
        if (strcmp(it->id, data.id) == 0) return true;
    }
    
    return false;
}

void insert(Data &data, HashTable *ht) {
    int pos = atoi(data.id) % 10000;
    ht->table[pos].push_back(data);
}

　　手写实现一般会比STL的要快，比如还有手写堆和优先队列。如果题目时间不是卡的很紧，还是用STL吧，毕竟太方便了。