Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Solution {
- public:
- vector<vector<string>> suggestedProducts(vector<string>& A, string searchWord) {
- auto it = A.begin();
- sort(A.begin(), A.end());
- vector<vector<string>> res;
- string cur = "";
- for (char c : searchWord) {
- cur += c;
- vector<string> suggested;
- it = lower_bound(it, A.end(), cur);
- for(int i = 0; i < 3 && it + i != A.end(); i++) {
- string& s = *(it + i);
- if(s.find(cur)) break;
- suggested.push_back(s);
- }
- res.push_back(suggested);
- }
- return res;
- }
- };
Add Comment
Please, Sign In to add comment