809_情感丰富的文字

难度:中等

题目

有时候人们会用重复写一些字母来表示额外的感受,比如 "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 。

提示

  • 0 <= len(S) <= 100。
  • 0 <= len(words) <= 100。
  • 0 <= len(words[i]) <= 100。
  • S 和所有在 words 中的单词都只由小写字母组成。

解题

根据题目意思,判断为可扩张的条件是扩张后字母数量需要大于等于三个

所以可以先对 S 进行统计,按顺序记录出现的字母和出现的个数,如 heeellooo 可以记录为 helo-1323,即 h 出现 1 次,e 出现 3 次,l 出现 2 次, o 出现 3 次。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class temp{
String key;
List<Integer> counts;

public temp(String s){
StringBuilder sb = new StringBuilder();
counts = new ArrayList<>();
char[] chars = s.toCharArray();
int n = chars.length;
int prev = -1;
for (int i=0; i<n; i++){
if (i==n-1 || chars[i]!=chars[i+1]){
sb.append(chars[i]);
counts.add(i-prev);
prev = i;
}
}
key = sb.toString();
}
}

然后对每个单词进行统计,首先判断出现的字母是否相同,不相同直接跳过。出现的字母相同后,比较每个字母出现的次数,这里需要满足:如果出现次数小于 3 次,那么单词里和句子里的次数都要相同;如果出现的次数大于 3 次,那么句子里的次数要超过单词里的次数,否则跳过该单词。

1
2
3
4
5
6
7
8
9
10
11
f (!S.key.equals(T.key)) {
continue;
}
for (int i = 0; i < S.counts.size(); i++) {
int count1 = S.counts.get(i);
int count2 = T.counts.get(i);
if (count1 < 3 && count1 != count2 || count1 < count2) {
continue search;
}
}
ans++;
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
class Solution {
public int expressiveWords(String s, String[] words) {
temp S = new temp(s);
int ans = 0;
search: for (String word : words) {
temp T = new temp(word);
if (!S.key.equals(T.key)) {
continue;
}
for (int i = 0; i < S.counts.size(); i++) {
int count1 = S.counts.get(i);
int count2 = T.counts.get(i);
if (count1 < 3 && count1 != count2 || count1 < count2) {
continue search;
}
}
ans++;
}
return ans;
}
}
class temp{
String key;
List<Integer> counts;

public temp(String s){
StringBuilder sb = new StringBuilder();
counts = new ArrayList<>();
char[] chars = s.toCharArray();
int n = chars.length;
int prev = -1;
for (int i=0; i<n; i++){
if (i==n-1 || chars[i]!=chars[i+1]){
sb.append(chars[i]);
counts.add(i-prev);
prev = i;
}
}
key = sb.toString();
}
}