Create print-words-vertically.cpp

Author: kamyu104

C++/print-words-vertically.cpp

@@ -0,0 +1,26 @@
+// Time:  O(n)
+// Space: O(n)
+
+class Solution {
+public:
+    vector<string> printVertically(string s) {
+        vector<string> result, words;
+        stringstream ss(s);
+        string word;
+        int max_len = 0;
+        while (ss >> word) {
+            words.emplace_back(word);
+            max_len = max(max_len, int(word.size()));
+        }
+        for (int i = 0; i < max_len; ++i) {
+            result.emplace_back();
+            for (const auto& word : words) {
+                result.back() += i < word.length() ? word[i] : ' ';
+            }
+            while (result.back().back() == ' ') {
+                result.back().pop_back();
+            }
+        }
+        return result;
+    }
+};