【思维题】B036_LC_字符串中的查找与替换（排序+暴力替换 / 反向构造）

一、Problem

对于某些字符串 S，我们将执行一些替换操作，用新的字母组替换原有的字母组（不一定大小相同）。

每个替换操作具有 3 个参数：起始索引 i，源字 x 和目标字 y。规则是如果 x 从原始字符串 S 中的位置 i 开始，那么我们将用 y 替换出现的 x。如果没有，我们什么都不做。

举个例子，如果我们有 S = “abcd” 并且我们有一些替换操作 i = 2，x = “cd”，y = “ffff”，那么因为 “cd” 从原始字符串 S 中的位置 2 开始，我们将用 “ffff” 替换它。

再来看 S = “abcd” 上的另一个例子，如果我们有替换操作 i = 0，x = “ab”，y = “eee”，以及另一个替换操作 i = 2，x = “ec”，y = “ffff”，那么第二个操作将不执行任何操作，因为原始字符串中 S[2] = ‘c’，与 x[0] = ‘e’ 不匹配。

所有这些操作同时发生。保证在替换时不会有任何重叠： S = “abc”, indexes = [0, 1], sources = [“ab”,“bc”] 不是有效的测试用例。

示例 1：

输入：S = "abcd", indexes = [0,2], sources = ["a","cd"], targets = ["eee","ffff"]
输出："eeebffff"
解释：
"a" 从 S 中的索引 0 开始，所以它被替换为 "eee"。
"cd" 从 S 中的索引 2 开始，所以它被替换为 "ffff"。
示例 2：

输入：S = "abcd", indexes = [0,2], sources = ["ab","ec"], targets = ["eee","ffff"]
输出："eeecd"
解释：
"ab" 从 S 中的索引 0 开始，所以它被替换为 "eee"。
"ec" 没有从原始的 S 中的索引 2 开始，所以它没有被替换。

提示：

0 <= indexes.length = sources.length = targets.length <= 100
0 < indexes[i] < S.length <= 1000
给定输入中的所有字符都是小写字母。

二、Solution

方法一：暴力替换

思路

• 判断 S 下标 indexes[i] 处开始的子字符串是否为 sources[i]，是的话替换为 targets[i]
• 需要注意的是直接改变S的话相关的下标对应的字符会发生改变。比较巧妙的方法是从后往前开始替换
• 但这题有点坑，indexes 非顺序，即使逆序遍历也可能会造成重叠替换，故先呀排个序

class Solution {
    public String findReplaceString(String s, int[] idxes, String[] ss, String[] ts) {
        StringBuilder sb = new StringBuilder(s);
        List<int[]> pp = new ArrayList<>();
        for (int i = 0; i < ss.length; i++)
            pp.add(new int[]{idxes[i], i});
        Collections.sort(pp, (e1, e2) -> e2[0] - e1[0]);

        for (int[] p : pp) {
            int start = p[0], i = p[1], len = ss[i].length();
            if (sb.substring(start, start+len).equals(ss[i]))
                sb.replace(start, start+len, ts[i]);
        }
        return sb.toString();
    }
}

复杂度分析

• 时间复杂度：$O(...)$，
• 空间复杂度：$O(...)$，

---

方法二：反向构造

思路

如果能在 s 中匹配 sources[i] 的话，那么记录此时目标串的位置，以便能在后面快速找到

class Solution {
public:
    string findReplaceString(string s, vector<int>& is, vector<string>& ss, vector<string>& ts) {
    	int n = s.size(), m = ss.size();
    	unordered_map<int, int> mp;
        for (int i = 0; i < m; i++) {
        	if (s.substr(is[i], ss[i].size()) == ss[i])
        		mp[is[i]] = i;
        }
        string ans = "";
        for (int i = 0; i < n;) {
        	if (mp.find(i) != mp.end()) {
        		ans += ts[mp[i]];
        		i += ss[mp[i]].size();
        	} else {
        		ans += s[i];
        		i++;
        	}
        }
        return ans;
    }
};

复杂度分析

• 时间复杂度：$O(n)$，
• 空间复杂度：$O(n)$，