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 [e-mail]
N 1234567890 [e-mail]
N 1234567890 [e-mail]
L 1234567890 myQQ@qq
L 1234567890 [e-mail]

輸出樣例:

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吧，畢竟太方便了。