Wordle solver.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "15b81476",
   "metadata": {},
   "outputs": [],
   "source": [
    "import json\n",
    "import ijson\n",
    "\n",
    "with open('allowed_words.txt') as f:\n",
    "    allowed_words = f.read().splitlines()\n",
    "                                \n",
    "with open('freq_map.json') as f:\n",
    "    freq_map = json.load(f)\n",
    "    \n",
    "possible_pattern = {}\n",
    "with open('possible_pattern.json') as f:\n",
    "    parser = ijson.parse(f)\n",
    "    for prefix, event, value in parser:\n",
    "        if event == \"start_map\" and prefix != \"\":\n",
    "            possible_pattern[prefix] = {}\n",
    "        elif event == \"string\":\n",
    "            possible_pattern[prefix[0:5]][prefix[6:11]] = value"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c782ae5f",
   "metadata": {},
   "outputs": [],
   "source": [
    "def hardMode(possible_words, guess_num, max_p):\n",
    "    total_p = 0\n",
    "    for i in possible_words:\n",
    "        total_p += freq_map[i]\n",
    "    \n",
    "    best_guess = \" \"*5\n",
    "    best_expect = 7\n",
    "    if total_p == 0:\n",
    "        best_expect = 7\n",
    "    else:\n",
    "        best_expect = max_p/total_p\n",
    "    \n",
    "    for guess in possible_words:\n",
    "        if guess_num < 6:\n",
    "            ## build the freq dict\n",
    "            freq_dict = {}\n",
    "            for answer in possible_words:\n",
    "                result = possible_pattern[guess][answer]\n",
    "                if result in freq_dict:\n",
    "                    freq_dict[result].append(answer)\n",
    "                else:\n",
    "                    freq_dict[result] = [answer]\n",
    "            \n",
    "            cum_p = 0\n",
    "            expect = 0\n",
    "            for k, v in freq_dict.items():\n",
    "                if total_p == 0:\n",
    "                    min_expect = expect\n",
    "                else:\n",
    "                    min_expect = expect + 1 - cum_p/total_p\n",
    "                    \n",
    "                if min_expect > best_expect:\n",
    "                    expect = min_expect\n",
    "                    break\n",
    "                \n",
    "                if k == \"GGGGG\":\n",
    "                    P = freq_map[v[0]]\n",
    "                    G = v[0];\n",
    "                    E = 0;\n",
    "                elif len(v) == 1:\n",
    "                    P = freq_map[v[0]]\n",
    "                    G = v[0];\n",
    "                    E = 1;\n",
    "                elif len(v) == 2:\n",
    "                    P = freq_map[v[0]] + freq_map[v[1]]\n",
    "                    if P == 0:\n",
    "                        G = v[0]\n",
    "                        E = 3/2\n",
    "                    elif freq_map[v[0]] > freq_map[v[1]]:\n",
    "                        G = v[0];\n",
    "                        E = 1+freq_map[v[1]]/P;\n",
    "                    else:\n",
    "                        G = v[1];\n",
    "                        E = 1+freq_map[v[0]]/P;\n",
    "                else:\n",
    "                    P, G, E = hardMode(v, guess_num+1, (best_expect-expect)*total_p)\n",
    "                    \n",
    "                cum_p += P\n",
    "                if total_p == 0:\n",
    "                    expect += len(v)/len(possible_words) * (E+1)\n",
    "                else:\n",
    "                    expect += P/total_p * (E+1)\n",
    "        else:\n",
    "            if total_p == 0:\n",
    "                expect = 1 + 1-1/len(possible_words)\n",
    "            else:\n",
    "                expect = 1 + 1-freq_map[guess]/total_p\n",
    "            \n",
    "        if expect < best_expect:\n",
    "            best_guess = guess\n",
    "            best_expect = expect\n",
    "\n",
    "    return total_p, best_guess, best_expect"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4ef81086",
   "metadata": {},
   "outputs": [],
   "source": [
    "possible_words = allowed_words\n",
    "\n",
    "print(\"Example: ROATE 🟨⬛🟩⬛⬛ input: roate YBGBB\")\n",
    "\n",
    "guess = ' '*11\n",
    "\n",
    "num = 0\n",
    "while not (guess[6:11] == \"GGGGG\"):\n",
    "    num += 1\n",
    "    next_possible_words = []\n",
    "\n",
    "    guess = input()\n",
    "    for char in guess:\n",
    "        if char == 'G':\n",
    "            print(\"🟩\", end='')\n",
    "        elif char == 'Y':\n",
    "            print(\"🟨\", end='')\n",
    "        elif char == 'B':\n",
    "            print(\"⬛\", end='')\n",
    "        else:\n",
    "            print(char, end='')\n",
    "    print()\n",
    "\n",
    "    for word in possible_words:\n",
    "        if possible_pattern[guess[0:5]][word] == guess[6:11]:\n",
    "            next_possible_words.append(word)\n",
    "    possible_words = next_possible_words\n",
    "    \n",
    "    print(\"# of possible solutions :\", len(possible_words))\n",
    "    \n",
    "    P, G, E = hardMode(possible_words, num+1, 7)\n",
    "    print(G, E)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "de6fbc94",
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "## finding start word\n",
    "\n",
    "TOTAL_P = 0.079688853104321\n",
    "\n",
    "P, G, E = hardMode(allowed_words, 1, 7*TOTAL_P)\n",
    "print(G, E)"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "d3d1920a",
   "metadata": {},
   "source": [
    "## entropy method\n",
    "* total entropy: 9.243309925972358\n",
    "* top 5 start word:\n",
    "    * ('tares', 5.96463766256561)\n",
    "    * ('tears', 5.907751486245412)\n",
    "    * ('tries', 5.90176559468474)\n",
    "    * ('tores', 5.900578491569622)\n",
    "    * ('stare', 5.855530842375718)\n",
    "    \n",
    "## brute force method\n",
    "* ('tears', 2.8521390723896127)\n",
    "* ('tares', 2.85717712365544)\n",
    "* ('taser', 2.874560888006384)\n",
    "* ('salet', 2.8937745301017723)\n",
    "\n",
    "tries\n",
    "tores\n",
    "stare\n",
    "\n",
    "roate\n",
    "raise\n",
    "\n",
    "crane"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "200cde41",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}