题目描述:
有时候人们会用重复写一些字母来表示额外的感受,比如 "hello" -> "heeellooo"
, "hi" -> "hiii"
。我们将相邻字母都相同的一串字符定义为相同字母组,例如:"h", "eee", "ll", "ooo"。
对于一个给定的字符串 S ,如果另一个单词能够通过将一些字母组扩张从而使其和 S 相同,我们将这个单词定义为可扩张的(stretchy)。扩张操作定义如下:选择一个字母组(包含字母 c
),然后往其中添加相同的字母 c
使其长度达到 3 或以上。
例如,以 "hello" 为例,我们可以对字母组 "o" 扩张得到 "hellooo",但是无法以同样的方法得到 "helloo" 因为字母组 "oo" 长度小于 3。此外,我们可以进行另一种扩张 "ll" -> "lllll" 以获得 "helllllooo"。如果 s = "helllllooo"
,那么查询词 "hello" 是可扩张的,因为可以对它执行这两种扩张操作使得 query = "hello" -> "hellooo" -> "helllllooo" = s
。
输入一组查询单词,输出其中可扩张的单词数量。
示例:
输入: s = "heeellooo" words = ["hello", "hi", "helo"] 输出:1 解释: 我们能通过扩张 "hello" 的 "e" 和 "o" 来得到 "heeellooo"。 我们不能通过扩张 "helo" 来得到 "heeellooo" 因为 "ll" 的长度小于 3 。
class Solution {
public:
int expressiveWords(string s, vector<string>& words) {
int n = words.size();
int a[26]{0};
for(int i = 0;i < s.size();i++){ //取得s的所有字符
a[s[i] - 'a']++;
}
int ans = 0;
for(int i = 0;i < n;i++){
int w[26]{0};
int isWord = 1;
for(auto t : words[i]){ //如果和s中的字符有不同的直接跳过到下一字符串
if(a[t - 'a'] == 0){isWord = 0;break;}
w[t - 'a']++;
}
if(isWord == 0)continue;
for(int fast = 0,slow = 0;fast < s.size() || slow < words[i].size();){
int ns = 0;
int nw = 0;
if(s[fast] != words[i][slow]){isWord = 0;break;} //如果字符不同则说明两都顺序不同无法扩展
else{
char same = s[fast];
while(s[fast] == same){fast++;ns++;} //统计相同的字符个数
while(words[i][slow] == same){slow++;nw++;}
if(ns == nw){continue;} //个数相同则继续比较下面字符
if(ns < nw){isWord = 0;break;} //如果s中字符少 则无法扩展
else if(ns > nw && ns == 2){isWord = 0;break;} //如果s中字符数量多且恰好为2个则无法扩展
}
}
ans += isWord; //将本字符加入答案
}
return ans;
}
};