diff --git a/src/algorithms/string.py b/src/algorithms/string.py index 658439e..cbdad04 100644 --- a/src/algorithms/string.py +++ b/src/algorithms/string.py @@ -30,10 +30,7 @@ def word_frequency(text: str) -> dict[str, int]: words = text.lower().split() frequency = {} for word in words: - if word in frequency: - frequency[word] += 1 - else: - frequency[word] = 1 + frequency[word] = frequency.get(word, 0) + 1 return frequency