Formatted question description: https://leetcode.ca/all/1324.html

1324. Print Words Vertically (Medium)

Given a string s. Return all the words vertically in the same order in which they appear in s.
Words are returned as a list of strings, complete with spaces when is necessary. (Trailing spaces are not allowed).
Each word would be put on only one column and that in one column there will be only one word.

 

Example 1:

Input: s = "HOW ARE YOU"
Output: ["HAY","ORO","WEU"]
Explanation: Each word is printed vertically. 
 "HAY"
 "ORO"
 "WEU"

Example 2:

Input: s = "TO BE OR NOT TO BE"
Output: ["TBONTB","OEROOE","   T"]
Explanation: Trailing spaces is not allowed. 
"TBONTB"
"OEROOE"
"   T"

Example 3:

Input: s = "CONTEST IS COMING"
Output: ["CIC","OSO","N M","T I","E N","S G","T"]

 

Constraints:

  • 1 <= s.length <= 200
  • s contains only upper case English letters.
  • It's guaranteed that there is only one space between 2 words.

Related Topics:
String

Solution 1.

// OJ: https://leetcode.com/problems/print-words-vertically/
// Time: O(S)
// Space: O(S)
class Solution {
public:
    vector<string> printVertically(string s) {
        vector<string> v, ans;
        stringstream ss(s);
        string word;
        int maxLen = 0;
        while (ss >> word) {
            v.push_back(word);
            maxLen = max(maxLen, (int)word.size());
        }
        for (int i = 0; i < maxLen; ++i) {
            string word;
            for (int j = 0; j < v.size(); ++j)
                word += i < v[j].size() ? v[j][i] : ' ';
            while (word.back() == ' ') word.pop_back();
            ans.push_back(word);
        }
        return ans;
    }
};

Java

  • class Solution {
        public List<String> printVertically(String s) {
            String[] array = s.split(" ");
            int wordsCount = array.length;
            int[] lengths = new int[wordsCount];
            int maxLength = 0;
            for (int i = 0; i < wordsCount; i++) {
                int length = array[i].length();
                lengths[i] = length;
                maxLength = Math.max(maxLength, length);
            }
            List<StringBuffer> stringBufferList = new ArrayList<StringBuffer>();
            for (int i = 0; i < maxLength; i++)
                stringBufferList.add(new StringBuffer());
            for (int i = 0; i < wordsCount; i++) {
                String word = array[i];
                int length = word.length();
                for (int j = 0; j < length; j++)
                    stringBufferList.get(j).append(word.charAt(j));
                for (int j = length; j < maxLength; j++)
                    stringBufferList.get(j).append(' ');
            }
            for (int i = 0; i < maxLength; i++) {
                StringBuffer sb = stringBufferList.get(i);
                while (sb.charAt(sb.length() - 1) == ' ')
                    sb.deleteCharAt(sb.length() - 1);
                stringBufferList.set(i, sb);
            }
            List<String> verticalList = new ArrayList<String>();
            for (int i = 0; i < maxLength; i++)
                verticalList.add(stringBufferList.get(i).toString());
            return verticalList;
        }
    }
    
  • // OJ: https://leetcode.com/problems/print-words-vertically/
    // Time: O(S)
    // Space: O(S)
    class Solution {
    public:
        vector<string> printVertically(string s) {
            vector<string> v, ans;
            stringstream ss(s);
            string word;
            int maxLen = 0;
            while (ss >> word) {
                v.push_back(word);
                maxLen = max(maxLen, (int)word.size());
            }
            for (int i = 0; i < maxLen; ++i) {
                string word;
                for (int j = 0; j < v.size(); ++j)
                    word += i < v[j].size() ? v[j][i] : ' ';
                while (word.back() == ' ') word.pop_back();
                ans.push_back(word);
            }
            return ans;
        }
    };
    
  • # 1324. Print Words Vertically
    # https://leetcode.com/problems/print-words-vertically/
    
    class Solution:
        def printVertically(self, s: str) -> List[str]:
            words = s.split(" ")
            m = len(max(words, key=len))
            n = len(words)
            res = ["" for _ in range(m)]
            
            for i,x in enumerate(words):
                for j in range(m):
                    if j >= len(x):
                        res[j] += " "
                    else:
                        res[j] += x[j]
            
            for i in range(m):
                res[i] = res[i].rstrip()
                    
            
            return res
            
    

All Problems

All Solutions