当前位置: 代码迷 >> 综合 >> leetcode 5316. 竖直打印单词(C++)
  详细解决方案

leetcode 5316. 竖直打印单词(C++)

热度:31   发布时间:2024-01-26 05:02:57.0

给你一个字符串 s。请你按照单词在 s 中的出现顺序将它们全部竖直返回。
单词应该以字符串列表的形式返回,必要时用空格补位,但输出尾部的空格需要删除(不允许尾随空格)。
每个单词只能放在一列上,每一列中也只能有一个单词。

 

示例 1:

输入:s = "HOW ARE YOU"
输出:["HAY","ORO","WEU"]
解释:每个单词都应该竖直打印。 "HAY""ORO""WEU"

示例 2:

输入:s = "TO BE OR NOT TO BE"
输出:["TBONTB","OEROOE","   T"]
解释:题目允许使用空格补位,但不允许输出末尾出现空格。
"TBONTB"
"OEROOE"
"   T"

示例 3:

输入:s = "CONTEST IS COMING"
输出:["CIC","OSO","N M","T I","E N","S G","T"]

 

提示:

  • 1 <= s.length <= 200
  • s 仅含大写英文字母。
  • 题目数据保证两个单词之间只有一个空格。

C++

class Solution {
public:vector<string> printVertically(string s) {vector<string> res;int n=s.size();vector<string> tmp;int start=0;int len=0;for(int i=0;i<n;i++){if(' '==s[i]){len=max(len,i-start);tmp.push_back(s.substr(start,i-start));start=i+1;}}tmp.push_back(s.substr(start));len=max(len,n-start);for(int k=0;k<len;k++){string str;for(int i=0;i<tmp.size();i++){if(k<tmp[i].size()){str+=tmp[i][k];}else{str+=' ';}}int num=str.size();for(int i=str.size()-1;i>=0;i--){if(str[i]!=' '){num=i+1;break;}}res.push_back(str.substr(0,num));}return res;}
};