题解 | #E 奏绝#

E 奏绝

显然在计算m次答案时无法进行遍历,所以要预处理一些东西。

用cnt0[i]和cnt1[i]表示从1到i中0的个数和1的个数,s0[i]和s1[i]表示从1到i中0的下标总和与1的下标总和,ans[i]表示1到i总影响值

更新上述列表是简单的,遍历一边序列即可全部更新完成。

最后是计算答案,对于每一个l, r,答案首先为ans[r] - ans[l - 1],不过l前面的数仍然对剩余的答案有影响,所以要减去l前面的0与l至r中的1的下标的组合,以及l前面的1与l~r中的0的下标的组合。

import sys, random
input = lambda: sys.stdin.readline().rstrip("\r\n")
mii = lambda: map(int, input().split())
mod = 998244353

n, m = mii()
c = input()

cnt0, cnt1 = [0] * (n + 1), [0] * (n + 1)
ans = [0] * (n + 1)
s0, s1 = [0] * (n + 1), [0] * (n + 1)

for i in range(1, n + 1):
    cnt0[i], cnt1[i] = cnt0[i - 1], cnt1[i - 1]
    s0[i], s1[i] = s0[i - 1], s1[i - 1]
    if c[i - 1] == '0':
        cnt0[i] += 1
        s0[i] += i
        ans[i] = (ans[i - 1] + cnt1[i - 1] * i - s1[i - 1]) % mod
    else: 
        cnt1[i] += 1
        s1[i] += i
        ans[i] = (ans[i - 1] + cnt0[i - 1] * i - s0[i - 1]) % mod
    
for _ in range(m):
    l, r = mii()
    tmp1 = (s1[r] - s1[l - 1]) * cnt0[l - 1] - s0[l - 1] * (cnt1[r] - cnt1[l - 1])
    tmp0 = (s0[r] - s0[l - 1]) * cnt1[l - 1] - s1[l - 1] * (cnt0[r] - cnt0[l - 1])
    print((ans[r] - ans[l - 1] - tmp1 - tmp0 + mod) % mod)
    
    
    
全部评论

相关推荐

02-01 12:05
复旦大学 Java
腾讯的提前批大概率应该是没有笔试的,但是这个时候有相当部分的同学简历估计都没有准备好,没准备好的同学也不用急,大部分都是3月之后开,这个时候开的绝大多数都是神仙打架,问的东西也比较难,打算投递的同学也多看下计算机网络和操作系统,腾讯对这部分的知识问的比较多。另外多刷下牛客的热门题库,刷题注意刷ACM模式,和牛客的周赛题,腾讯有的部门会从这里面出原题。我是@程序员花海关注我,带你了解更多校招资讯!
程序员花海:还没有来得及准备的同学可以看下学习路线:https://www.nowcoder.com/discuss/824693499982315520?sourceSSR=users算法题:https://www.nowcoder.com/feed/main/detail/20e7a999fa04485b88340a274411ca0d?sourceSSR=users八股文:https://www.nowcoder.com/discuss/833102362771251200?sourceSSR=users简历书写方式:https://www.nowcoder.com/discuss/839907820706205696?sourceSSR=users都是以前在牛客发的文章~
软开人,秋招你打算投哪些...
点赞 评论 收藏
分享
评论
12
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务