dongar
414380740f
This is prep for having additional solvers that can be trivially swapped in and out.
88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
#!/usr/bin/python
|
|
|
|
import string
|
|
from collections import defaultdict
|
|
|
|
import constraints
|
|
import dictionary
|
|
import game
|
|
|
|
|
|
class SimilaritySolver(object):
|
|
@staticmethod
|
|
def _analyze_dictionary(words):
|
|
# Word Position, Letter: Count
|
|
position_matches = [
|
|
defaultdict(int),
|
|
defaultdict(int),
|
|
defaultdict(int),
|
|
defaultdict(int),
|
|
defaultdict(int),
|
|
]
|
|
|
|
# Letter: Count
|
|
word_matches = defaultdict(int)
|
|
|
|
for word in words:
|
|
unique_letters = set(word)
|
|
for letter in unique_letters:
|
|
word_matches[letter] += 1
|
|
|
|
for index in range(len(word)):
|
|
letter = word[index]
|
|
assert letter in string.ascii_lowercase
|
|
position_matches[index][letter] += 1
|
|
|
|
return position_matches, word_matches
|
|
|
|
@staticmethod
|
|
def _score_word(position_matches, word_matches, word):
|
|
result = 0
|
|
|
|
# Green Score
|
|
for index in range(len(word)):
|
|
letter = word[index]
|
|
green_score = position_matches[index][letter]
|
|
result += 10 * green_score
|
|
|
|
# Yellow Score
|
|
for letter in set(word):
|
|
yellow_score = word_matches[letter]
|
|
result += yellow_score
|
|
|
|
return result
|
|
|
|
def scored_words(self, wordle_dictionary=dictionary.words()):
|
|
# word: score
|
|
scored_words = dict()
|
|
|
|
position_matches, word_matches = self._analyze_dictionary(wordle_dictionary)
|
|
|
|
for word in wordle_dictionary:
|
|
scored_words[word] = self._score_word(position_matches, word_matches, word)
|
|
|
|
result = list(wordle_dictionary)
|
|
result.sort(key=lambda word: scored_words[word], reverse=True)
|
|
|
|
return [(word, scored_words[word]) for word in result]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
solve = SimilaritySolver()
|
|
|
|
consts = constraints.Constraints()
|
|
consts.update_graded(
|
|
"cares",
|
|
[
|
|
game.LetterResult.INCORRECT,
|
|
game.LetterResult.INCORRECT,
|
|
game.LetterResult.INCORRECT,
|
|
game.LetterResult.INCORRECT,
|
|
game.LetterResult.INCORRECT,
|
|
],
|
|
)
|
|
|
|
valid_words = consts.filter_words(dictionary.words())
|
|
|
|
print("Top: ", solve.scored_words(valid_words)[:5])
|