648. 单词替换

力扣原题链接(点我直达)

在英语中,我们有一个叫做 词根(root)的概念,它可以跟着其他一些词组成另一个较长的单词——我们称这个词为 继承词(successor)。例如,词根an,跟随着单词 other(其他),可以形成新的单词 another(另一个)。

现在,给定一个由许多词根组成的词典和一个句子。你需要将句子中的所有继承词词根替换掉。如果继承词有许多可以形成它的词根,则用最短的词根替换它。

你需要输出替换之后的句子。

示例 1:

输入: dict(词典) = ["cat", "bat", "rat"]
sentence(句子) = "the cattle was rattled by the battery"
输出: "the cat was rat by the bat"

注:

  1. 输入只包含小写字母。
  2. 1 <= 字典单词数 <=1000
  3. 1 <= 句中词语数 <= 1000
  4. 1 <= 词根长度 <= 100
  5. 1 <= 句中词语长度 <= 1000

第一版,自己写的,比较慢

执行用时 :384 ms, 在所有 cpp 提交中击败了19.37%的用户

内存消耗 :112.3 MB, 在所有 cpp 提交中击败了20.75%的用户

bool static compare(const string& a, const string& b) {
    return a.size() < b.size();

}

string replaceWords(vector<string>& dict, string sentence) {

    sort(dict.begin(), dict.end(), compare);
    string temp,result;
    for (unsigned i = 0; i < sentence.size(); ++i) {
        temp = "";
        while (sentence[i] != ' ' && i < sentence.size()) {
            temp += sentence[i++];
        }

        for (auto& d : dict) {
            if (temp.substr(0, d.size()) == d) {
                temp = d;
                break;
            }

        }
        result += temp;
        result += " ";

    }

    return result.substr(0, result.size() - 1);

}

第二版,改进了一点

执行用时 :108 ms, 在所有 cpp 提交中击败了52.57%的用户

内存消耗 :19.5 MB, 在所有 cpp 提交中击败了92.45%的用户

string replaceWords(vector<string>& dict, string sentence) {

    unordered_set<string> unst(dict.begin(), dict.end());
    string temp,result;
    for (unsigned i = 0; i < sentence.size(); ++i) {
        temp = "";
        while (sentence[i] != ' ' && i < sentence.size()) {
            temp += sentence[i];
            if (unst.find(temp) != unst.end()) {//此时已经找到了前缀了
                break;
            }
            i++;
        }
        result += temp;
        result += " ";
        while (sentence[i] != ' ' && i < sentence.size())
        {
            ++i;
        }//将整个单词跨过去。直到遇到空格  

    }

    return result.substr(0, result.size() - 1);

}