每日leetcode 466 2020-03-17

2020-04-22  本文已影响0人  五道口的程序狐

已经发布在了leetcode中文版里面。

题目

466. 统计重复个数

由 n 个连接的字符串 s 组成字符串 S,记作 S = [s,n]。例如,["abc",3]=“abcabcabc”。

如果我们可以从 s2中删除某些字符使其变为 s1,则称字符串 s1可以从字符串 s2 获得。例如,根据定义,"abc" 可以从 “abdbec” 获得,但不能从 “acbbe” 获得。

现在给你两个非空字符串 s1 和 s2(每个最多 100 个字符长)和两个整数 0 ≤ n1 ≤ 106 和 1 ≤ n2 ≤ 106。现在考虑字符串 S1 和 S2,其中 S1=[s1,n1]S2=[s2,n2]

请你找出一个可以满足使[S2,M]S1 获得的最大整数 M 。

示例:

输入:
s1 ="acb",n1 = 4
s2 ="ab",n2 = 2

返回:
2

总体思路

记录下来每个s1对应的位置,匹配的结果。每一次如果要求的匹配已经在数据中,就看可以匹配多少组。

比如,【baab, 40】里面要匹配【baba, 2】,那我就先找到应该匹配哪几位:

因此,我们就可以记录下来,每一次从s1的某一个位置进行匹配,会匹配到哪个地方。

匹配位置

匹配位置是说,使用一个数组/map,i对应的值为记录从s1的第i位开始,往后走到哪里可以匹配一个完整的s2。

比如:s1="baab", s2="baba",我用来记忆的map叫做memo,这样的话memo[0] = 6,表示从s1的下标为0的地方,往右走到下标为6的地方可以匹配完一个s2

如果是求memo[2]的话,那结果就等于10,代表下标为2~9的部分(ba【abbaabba】ab)可以匹配一个完整的s2。

因此,我这里设计了一个函数可以实现上述的功能:

int checkWhere(int n, string &s1, int len1, string &s2, int len2) {
    for (int i = 0; i < len2; i++) {
        int n_target = n + len1;
        bool found = false;
        for (; n < n_target; n++) {
            if (s1[n % len1] == s2[i]) {
                n++;
                found = true;
                break;
            }
        }
        if (!found) {
            return -1;
        }
    }
    return n;
}

找到循环体

比如刚才的例子,【baab, 40】里面要匹配【baba, 2】

到第二次寻找的时候,就是从下标为2开始,到下标为2结束。

再往后一次也是一样的,这样就构成了循环结构。

我在这里使用了一个循环结构的检测,记录的时候不仅记录应该到哪个地方,而且记录了到这个地方的时候我匹配完了几个s2。

也就是说,memo[0] = [6, 1],memo[2] = [10, 2]。

问题来了:

代码

class Solution {
public:
    int checkWhere(int n, string &s1, int len1, string &s2, int len2, map<int, pair<int, int>>& memo) {
        for (int i = 0; i < len2; i++) {
            int n_target = n + len1;
            bool found = false;
            for (; n < n_target; n++) {
                if (s1[n % len1] == s2[i]) {
                    n++;
                    found = true;
                    break;
                }
            }
            if (!found) {
                return -1;
            }
        }
        return n;
    }

    int getMaxRepetitions(string s1, int n1, string s2, int n2) {
        map<int, pair<int, int>> memo;
        int now_n = 0;  // from the nth letter of s1, to which will have one s2
        int len1 = s1.size(), len2 = s2.size();
        int max_n = len1 * n1;
        int now_num_s2 = 0;
        int n, temp;
        while (now_n < max_n) {
            n = now_n % len1;
            if (memo.find(n) != memo.end()) {
                // now already found
                // 之前用了now_n这么多个,也就是用了now_n//len1这么多个s1,并且找到了now_num_s2这么多个s2
                // 现在需要做什么?现在需要跳回到data[n][0]这个位置,并且可以得到now_num_s2 - memo[n][1]这么多个s2
                // 需要用到多少s1呢?
                int need_n = n / len1 * len1 + memo[n].first;
                while (need_n % len1 != n) {
                    need_n = need_n / len1 * len1 + memo[need_n % len1].first;
                }
                // 现在到了need_n这么多位的s1,也就是用了need_n/len1这么多个s1,就拿到了now_num_s2 - memo[n][1]这么多个s2
                // 还剩下max_n - now_n这么多个可以挥霍
                int add_whole = (max_n - now_n) / (need_n / len1 * len1);
                now_num_s2 += add_whole * (now_num_s2 - memo[n].second + 1);
                now_n += add_whole * (need_n / len1 * len1);
                while (now_n < max_n) {
                    temp = memo[now_n % len1].first;
                    now_n = now_n / len1 * len1 + temp;
                    if (now_n > max_n) {
                        break;
                    }
                    now_num_s2 += 1;
                }
                return now_num_s2 / n2;
            }
            temp = checkWhere(n, s1, len1, s2, len2, memo);
            if (temp == -1) {
                return 0;
            }
            now_n = now_n / len1 * len1 + temp;
            if (now_n > max_n) {
                break;
            }
            now_num_s2 += 1;
            memo[n] = pair<int, int>(temp, now_num_s2);
        }
        return now_num_s2 / n2;
    }
};
上一篇下一篇

猜你喜欢

热点阅读