ALDS1_4_C 辞書をC++で解いてみた[AIZU ONLINE JUDGE]

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

int const row = 1046527;
int const col = 14;

char hash_table[row][col];

long long get_char(char c) {
    if(c == 'A') {
        return 1;
    } else if(c == 'C') {
        return 2;
    } else if(c == 'G') {
        return 3;
    } else if(c == 'T') {
        return 4;
    }
    return 0;
}

long long get_key(char str[]) {
    long long sum = 0;
    long long p = 1;
    int len = strlen(str);
    for(long long i = 0; i < len; i++) {
        sum += p * get_char(str[i]);
        p *= 5;
    }
    return sum;
}

int h1(int key) { return key % row; }
int h2(int key) { return 1 + (key % (row - 1)); }

int find(char str[]) {
    long long key = get_key(str), h;
    for(int i = 0;; i++) {
        h = (h1(key) + i * h2(key)) % row;
        if(strcmp(hash_table[h], str) == 0) {
            return 1;
        } else if(strlen(hash_table[h]) == 0) {
            return 0;
        }
    }
}

int insert(char str[]) {
    long long key = get_key(str);
    long long h;
    for(int i = 0;; i++) {
        h = (h1(key) + i * h2(key)) % row;
        if(strcmp(hash_table[h], str) == 0) {
            return 1;
        } else if(strlen(hash_table[h]) == 0) {
            strcpy(hash_table[h], str);
            return 0;
        }
    }
}

int main() {
    int number;
	cin >> number;
    char cmd[20], str[20];
    for(int i = 0; i < row; i++) {
        hash_table[i][0] = '\0';
    }
	vector<bool> ans;
    for(int i = 0; i < number; i++) {
        cin >> cmd >> str;
        if(cmd[0] == 'i') {
            insert(str);
        } else {
            if(find(str)) {
				ans.push_back(true);
            } else {
				ans.push_back(false);
            }
        }
    }

	for (auto b : ans) {
		if (b) {
			cout << "yes" << endl;
		}
		else {
			cout << "no" << endl;
		}
	}

    return 0;
}