Create print-words-vertically.py

Author: kamyu104

Python/print-words-vertically.py

@@ -0,0 +1,13 @@
+# Time:  O(n)
+# Space: O(h)
+
+import itertools
+
+
+class Solution(object):
+    def printVertically(self, s):
+        """
+        :type s: str
+        :rtype: List[str]
+        """
+        return ["".join(c).rstrip() for c in itertools.izip_longest(*s.split(), fillvalue=' ')]