palperm2

Author: nickolasteixeira

Python/chapter01/1.4 - PalinPerm/Nick4.py

@@ -6,7 +6,7 @@
 '''
 
 import unittest
-
+import collections
 
 def palperm(string):
     '''
@@ -26,6 +26,19 @@ def palperm(string):
             count += 1
     return True
 
+def palperm2(string):
+    letter_frequencies = collections.defaultdict(int)
+    for letter in string.lower().replace(' ', ''):
+        letter_frequencies[letter] += 1
+    
+    odd_frequency_count = 0
+    for frequency in letter_frequencies.values():
+        if frequency % 2 == 0:
+            continue
+        odd_frequency_count += 1
+        if odd_frequency_count > 1:
+            return False
+    return True
 
 class Test(unittest.TestCase):
     def test1(self):
@@ -38,5 +51,15 @@ def test1(self):
         self.assertTrue(palperm(input_string3))
         self.assertTrue(palperm(input_string4))
 
+    def test2(self):
+        input_string ="Tact Coa"
+        input_string2 ="nick"
+        input_string3 ="saippuakivikauppias"
+        input_string4 = "iasppaukivikauppias"
+        self.assertTrue(palperm2(input_string))
+        self.assertFalse(palperm2(input_string2))
+        self.assertTrue(palperm2(input_string3))
+        self.assertTrue(palperm2(input_string4))
+
 if __name__ == '__main__':
     unittest.main()