leetcode809. 情感丰富的文字

有时候人们会用重复写一些字母来表示额外的感受，比如 “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, String[] words) { int n=S.length(),res=0; if (n==0) return 0; int[] jump=new int[n];//记录出现的多个连续重复字符的末尾位置 jump[n-1]=n-1; for(int i=n-2;i>=0;i--) { if(S.charAt(i)==S.charAt(i+1)) jump[i]=jump[i+1]; else jump[i]=i; } for(String string:words)//遍历words { int start=0,i=0; for(;i<string.length()&&start<n;i++)//遍历单词的每个字符 { if(string.charAt(i)==S.charAt(start))//相同字符 { int len=1; while (i+1<string.length()&&string.charAt(i)==string.charAt(i+1)) //找出后面相同字符的长度 { i++; len++; } int len2=jump[start]+1-start;//根据jump数组直接得出最后一个重复字符的位置 if(len2<=2&&len2!=len||len>len2) //当S中字符连续的长度小于2，不能任意匹配长度，必须和word连续的长度相同 break; start=jump[start]+1; }else break; } if(i==string.length()&&start==n) res++; } return res; } }