二分查找应用 | 判断字符串子序列

题目信息

给定字符串 s 和 t ，判断 s 是否为 t 的子序列。

字符串的一个子序列是原始字符串删除一些（也可以不删除）字符而不改变剩余字符相对位置形成的新字符串。（例如，"ace"是"abcde"的一个子序列，而"aec"不是）。

进阶：

如果有大量输入的 S，称作 S1, S2, ... , Sk 其中 k >= 10亿，你需要依次检查它们是否为 T 的子序列。在这种情况下，你会怎样改变代码？

进阶方法分析

得到字符“字典”

首先需要对原始字符串进行处理，将字符串中每个字符对应的 index 存储在 vector<int> base_str[26] 中，其中索引为对应的 26 个字符串。

得到 base_str 需要的时间复杂度为 O(N)。

int length_t = t.length();
for (int i = 0; i < length_t; i++) {
    base_str[t[i] - 'a'].push_back(i);
}

查找对应的字符位置

利用二分法查找对应的字符是否存在。

    //二分查找
    int binarySearch(const vector<int>& strs, int target) {
        int left = 0;
        int right = strs.size();
        while (left < right) {
            int mid = (left + right) >> 1;
            if (strs[mid] >= target) {
                right = mid;
            }
            else {
                left = mid + 1;
            }
        }
        return left;
    }

完整代码实现

#define LETTER 26
class Solution {
public:
    //每个base_str[i]中都是按照从小至大的顺序排序的，可以方便二分查找。
    vector<int> base_str[LETTER];
    //初始化base_str。
    void initStr() {
        for (int i = 0; i < LETTER; i++) {
            base_str[i].clear();
        }
    }
    //二分查找
    int binarySearch(const vector<int>& strs, int target) {
        int left = 0;
        int right = strs.size();
        while (left < right) {
            int mid = (left + right) >> 1;
            if (strs[mid] >= target) {
                right = mid;
            }
            else {
                left = mid + 1;
            }
        }
        return left;
    }

    bool isSubsequence(string s, string t) {
        initStr();
        int length_t = t.length();
        for (int i = 0; i < length_t; i++) {
            base_str[t[i] - 'a'].push_back(i);
        }

        int idx_t = 0;
        for (auto its : s) {
            int idx = its - 'a';
            //没有找到字符its，直接返回false
            if (base_str[idx].empty()) {
                return false;
            }

            int pos = binarySearch(base_str[idx], idx_t);
            //如果没有找到会返回最大 index + 1，也就是 vector 的大小。
            if (pos == base_str[idx].size()) {
                return false;
            }
            //找下一个，所以需要 +1
            idx_t = base_str[idx][pos] + 1;
        }
        return true;
    }
};