#include <iostream>
#include <windows.h>
#include <string>
#include <vector>

std::vector<uint64_t> keyValues;
std::vector<std::string> Keys;

void intree(std::string key, uint64_t num, uint64_t index)
{
	for (unsigned int i = 0; i < Keys.size(); i++)
		if (key == Keys[i])
		{
			keyValues[i] = num;
		}
	keyValues.insert(keyValues.begin() + index, num);
	Keys.insert(Keys.begin() + index, key);
}

int outtree(const char* key, int& invKey)
{
	for (unsigned int i = 0; i < Keys.size(); i++)
	{
		if (!strcmp(key, Keys.at(i).c_str()))
		{
			return keyValues.at(i);
		}
	}
	invKey = 1;
	printf("invalid key\n");
	return 0;
}

int main() {
	std::string input = "123456789";
	uint64_t p_num = 123456789, keyInd = 0;
	std::vector<uint64_t> num;
	std::vector<std::string> key;
	while (1)
	{
	beginning:
		std::cout << "Setup key: ";
		//std::cin >> input;
		if (input == "exit")
			break;
		key.push_back(input);
		//std::cin >> p_num;
		num.push_back(p_num);
		intree(key.at(keyInd), num[keyInd], keyInd);
		keyInd++;
	}
	while (1)
	{
	start:
		int invalidKey = 0;
		std::cout << "Enter key: ";
		std::string input = "";
		std::cin >> input;
		if (input == "new_key")
			goto beginning;
		else if (input == "exit")
			break;
		int result = 0;
		result = outtree(input.c_str(), invalidKey);
		if (invalidKey)
			goto start;
		printf("%u\n", result);
	}
	return 0;