寻找 520 题解

寻找 520

根据题意,我们需要考虑所有位置不同的 520 子序列的位置。根据分数我们有下面几种做法:

20 分的做法

由于至多只有 5 个问号,我们可以 枚举这些位置的取值然后从每一个位置出发 DFS 搜索找答案并去重即可。令 为 ? 的个数,则复杂度为

50 分的做法

现在整个字符串的长度有 的 DFS 显然是无法承受的。我们来考虑怎么优化这个 DFS。

我们考虑枚举 2 的位置,观察到对于每个 2,只有它前面的 5 和后面的 0 参与到答案的贡献来,而后面的 5 和前面 0 没有任何贡献。这就提示我们先对整个序列扫一遍,计算 5 和 0 个数的前缀和(注意有了 之后,我们计算区间内的和可以简单计算为 ),然后再扫一遍,根据 2 进行计数即可。

所以我们只需要枚举 ? 处的字符分别跑一遍上面的计数过程然后加起来就有答案了,整体复杂度

100 分的做法

我们考虑第二种做法,容易注意到,随着 2 的位置的不同,我们枚举出来的 520 子串是不重不漏的。这就提示我们小幅修改上面的 40 分做法就有新的答案:我们扫第一遍统计的时候,把 ? 当做 5 和 0 也统计进前缀和中,然后第二遍枚举的时候,当前字符为 2 或者 ? 都统计进答案就可以了。复杂度 ,实现的时候注意一下区间边界,统计不要有重复。

C++ 参考代码:

class Solution {
  using ll = long long;
  const static int SIZE = 1e6 + 10;
  const static int MOD = 998244353;

  int fives[SIZE], zeros[SIZE];

public:
  /**
   * 寻找所有的 520 子序列
   * @param S string字符串 牛牛准备的数字字符串
   * @return int整型
   */
  int findOccurrences(string S) {
    int n = S.size();

    for (int i = 1; i <= n; i++) {
      fives[i] = fives[i - 1];
      zeros[i] = zeros[i - 1];
      if (S[i - 1] == '5') {
        fives[i]++;
      } else if (S[i - 1] == '0') {
        zeros[i]++;
      } else if (S[i - 1] == '?') {
        fives[i]++;
        zeros[i]++;
      }
    }

    ll ans = 0;

    for (int i = 1; i <= n; i++) {
      if (S[i - 1] == '2' || S[i - 1] == '?') {
        ans = (ans + 1LL * fives[i - 1] * (zeros[n] - zeros[i]) % MOD) % MOD;
      }
    }

    return ans;
  }
};
全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务