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 中的單詞都只由小寫字母組成。
代碼
class Solution {
public:
int expressiveWords(string S, vector<string>& words) {
int res = 0;
for (auto it : words) {
if (match(S, it))
res++;
}
return res;
}
bool match(string s, string w) {
int sl = s.size();
int wl = w.size();
int i = 0, j = 0;
while (i < sl && j < wl) {
if (s[i] != w[j])return false;
int stag = 0;//用來計數當前字符在s中的個數
while ((i + stag) < sl && s[i + stag] == s[i])stag++;
i += stag;//更新s中i的下標
int wtag = 0;//用來計數當前字符在w中的個數
while ((j + wtag) < wl && w[j + wtag] == w[j])wtag++;
j += wtag;//更新w中j的下標
if (stag == wtag || (stag >= 3 && stag > wtag)) continue;//stag>wtag是防止這種情況"aaa"["aaaa"]這結果是false
return false;
}
return i == sl && j == wl;//都走到了最後說明正確。
}
};
int main() {
//vector<int>nums = { 1,2 };
string s = "heeellooo";
vector<string>words = { "hello", "hi", "helo" };
Solution S;
cout<<S.expressiveWords(s,words)<<endl;//結果爲1
}