day06: s/for loops/generators!

tfaughnan · tfaughnan · commit 9a27564b1cb4 · 2022-12-06T00:32:53.000-05:00
diff --git a/day06/1.py b/day06/1.py
@@ -6,10 +6,9 @@
 
 def main():
     data = sys.stdin.read().strip()
-    for i in range(len(data) - WINSIZ):
-        if len(set(data[i:i + WINSIZ])) == WINSIZ:
-            print(i + WINSIZ)
-            break
+    nchars = next(i + WINSIZ for i in range(len(data) - WINSIZ)
+                  if len(set(data[i:i + WINSIZ])) == WINSIZ)
+    print(nchars)
 
 if __name__ == '__main__':
     main()
diff --git a/day06/2.py b/day06/2.py
@@ -6,10 +6,9 @@
 
 def main():
     data = sys.stdin.read().strip()
-    for i in range(len(data) - WINSIZ):
-        if len(set(data[i:i + WINSIZ])) == WINSIZ:
-            print(i + WINSIZ)
-            break
+    nchars = next(i + WINSIZ for i in range(len(data) - WINSIZ)
+                  if len(set(data[i:i + WINSIZ])) == WINSIZ)
+    print(nchars)
 
 if __name__ == '__main__':
     main()
