83 lines
2.0 KiB
Python
83 lines
2.0 KiB
Python
#!/usr/bin/python
|
|
|
|
import string
|
|
from collections import defaultdict
|
|
|
|
import dictionary
|
|
|
|
|
|
def analyze_dictionary(wordle_dictionary):
|
|
# 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 wordle_dictionary:
|
|
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
|
|
|
|
|
|
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(wordle_dictionary=dictionary.words()):
|
|
# word: score
|
|
scored_words = dict()
|
|
|
|
position_matches, word_matches = analyze_dictionary(wordle_dictionary)
|
|
|
|
for word in wordle_dictionary:
|
|
scored_words[word] = 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__":
|
|
consts = constraints.Constraints()
|
|
consts.update_graded(
|
|
"cares",
|
|
[
|
|
solver.LetterResult.INCORRECT,
|
|
solver.LetterResult.INCORRECT,
|
|
solver.LetterResult.INCORRECT,
|
|
solver.LetterResult.INCORRECT,
|
|
solver.LetterResult.INCORRECT,
|
|
],
|
|
)
|
|
|
|
valid_words = consts.filter_words(dictionary.words())
|
|
|
|
print("Top: ", solver.scored_words(valid_words)[:5])
|