「句子」是一个用空格分隔单词的字符串。给你一个满足下述格式的句子 text :
句子的首字母大写
text 中的每个单词都用单个空格分隔。
请你重新排列 text 中的单词,使所有单词按其长度的升序排列。如果两个单词的长度相同,则保留其在原句子中的相对顺序。
请同样按上述格式返回新的句子。
示例 1:
输入:text = "Leetcode is cool"
输出:"Is cool leetcode"
解释:句子中共有 3 个单词,长度为 8 的 "Leetcode" ,长度为 2 的 "is" 以及长度为 4 的 "cool" 。
输出需要按单词的长度升序排列,新句子中的第一个单词首字母需要大写。
示例 2:
输入:text = "Keep calm and code on"
输出:"On and keep calm code"
解释:输出的排序情况如下:
"On" 2 个字母。
"and" 3 个字母。
"keep" 4 个字母,因为存在长度相同的其他单词,所以它们之间需要保留在原句子中的相对顺序。
"calm" 4 个字母。
"code" 4 个字母。
示例 3:
输入:text = "To be or not to be"
输出:"To be or to be not"
提示:
- text 以大写字母开头,然后包含若干小写字母以及单词间的单个空格。
- 1 <= text.length <= 10^5
主要思路:字符串分割和自定义排序(这个恶心点)
Code:
class Solution {
public:
typedef struct par
{
string str;
int pos;
}par;
void str_split(const std::string & src, const std::string & sep, std::vector<par> & vec_str)
{
std::string::size_type start = 0;
int i=0;
for(std::string::size_type end = src.find(sep, start); end != std::string::npos; end = src.find(sep, start))
{
if(end > start)
{
par p;
p.pos=i++;
p.str=src.substr(start, end - start);
vec_str.push_back(p);
}
start = end + sep.length();
}
if(start < src.length())
{
par p;
p.pos=i++;
p.str=src.substr(start, src.length() - start);
vec_str.push_back(p);
}
}
static bool cmpss(par s1,par s2)
{
if(s1.str.length()==s2.str.length())
return s1.pos<s2.pos;
return s1.str.size()<s2.str.size();
}
string arrangeWords(string text) {
vector<par>vec;
str_split(text," ",vec);
sort(vec.begin(),vec.end(),cmpss);
string res="";
for(int i=0;i<vec.size();i++)
{
if(i==0)
vec[i].str[0]=toupper( vec[i].str[0]);
else
vec[i].str[0]=tolower( vec[i].str[0]);
// cout<<vec[i]<<endl;
res+=vec[i].str;
if(i!=vec.size()-1)
{
res+=" ";
}
}
cout<<res<<endl;
return res;
}
};