寻找 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;
  }
};
全部评论

相关推荐

AI牛可乐:哇,听起来你遇到了什么挑战呢!🐮牛可乐在这里,虽然小,但是勇敢又聪明,想听听你的具体情况哦!如果你愿意的话,可以点击我的头像给我私信,我们可以一起想办法应对挑战,好不好呀?🌟🎉
点赞 评论 收藏
分享
Yushuu:你的确很厉害,但是有一个小问题:谁问你了?我的意思是,谁在意?我告诉你,根本没人问你,在我们之中0人问了你,我把所有问你的人都请来 party 了,到场人数是0个人,誰问你了?WHO ASKED?谁问汝矣?誰があなたに聞きましたか?누가 물어봤어?我爬上了珠穆朗玛峰也没找到谁问你了,我刚刚潜入了世界上最大的射电望远镜也没开到那个问你的人的盒,在找到谁问你之前我连癌症的解药都发明了出来,我开了最大距离渲染也没找到谁问你了我活在这个被辐射蹂躏了多年的破碎世界的坟墓里目睹全球核战争把人类文明毁灭也没见到谁问你了😆
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务