給定一個字元串 s 和一些長度相同的單詞 words。找出 s 中恰好可以由 words 中所有單詞串聯形成的子串的起始位置。註意子串要與 words 中的單詞完全匹配,中間不能有其他字元,但不需要考慮 words 中單詞串聯的順序。 示例 1:輸入: s = "barfoothefoobarman ...
給定一個字元串 s 和一些長度相同的單詞 words。找出 s 中恰好可以由 words 中所有單詞串聯形成的子串的起始位置。
註意子串要與 words 中的單詞完全匹配,中間不能有其他字元,但不需要考慮 words 中單詞串聯的順序。
示例 1:
輸入:
s = "barfoothefoobarman",
words = ["foo","bar"]
輸出:[0,9]
解釋:
從索引 0 和 9 開始的子串分別是 "barfoor" 和 "foobar" 。
輸出的順序不重要, [9,0] 也是有效答案。
示例 2:
輸入:
s = "wordgoodgoodgoodbestword",
words = ["word","good","best","word"]
輸出:[]
思路:首先為了方便比較,把原vector和用於存儲的目標vector都要排序,排序後就可以知道兩者是否完全相等了;剩下的就是暴力迴圈;
#include <iostream> #include <vector> #include <algorithm> using namespace std; vector<int> findSubstring(string s, vector<string>& words) { sort(words.begin(),words.end()); vector<int> ans; vector<string> temp; int n=words.size(); if(n==0) return ans; int len=words[0].length(); if(s.length()<n*len) return ans; for(int i=0;i<s.length()-n*len+1;i++) { int t=i; if(find(words.begin(),words.end(),s.substr(t,len))!=words.end()) { temp.push_back(s.substr(t,len)); t=t+len; } for(int j=1;j<n;j++) { if(find(words.begin(),words.end(),s.substr(t,len))!=words.end()) { temp.push_back(s.substr(t,len)); t=t+len; } else { temp.clear(); break; } } sort(temp.begin(),temp.end()); if(temp==words) ans.push_back(i); temp.clear(); } return ans; } int main() { string s="aaa"; vector<string> a={"a","a"}; vector<int> ans=findSubstring(s,a); std::cout << ans.size() << std::endl; //std::cout << ans[0]<<ans[1] << std::endl; return 0; }