結果

問題 No.2611 Count 01
コンテスト
ユーザー Mitarushi
提出日時 2024-01-07 11:03:11
言語 Nim
(2.2.0)
結果
WA  
実行時間 -
コード長 1,166 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,715 ms
コンパイル使用メモリ 66,176 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-28 03:29:50
合計ジャッジ時間 4,582 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import strutils, sequtils, deques

type stdinReader = object
    buf: Deque[string]

func newStdinReader(): stdinReader =
    result = stdinReader(buf: initDeque[string]())

proc readString(sr: var stdinReader): string =
    if sr.buf.len == 0:
        sr.buf = stdin.readLine.split.toDeque
    sr.buf.popFirst

proc readInt(sr: var stdinReader): int =
    readString(sr).parseInt

var reader = newStdinReader()
let (n, q) = (reader.readInt, reader.readInt)
let s = reader.readString
var sSeq = s.mapIt(int(it) - int('0'))

const MOD = 998244353

if n > 10000 or q > 10000:
    quit(0)

for _ in 0..<q:
    let t = reader.readInt
    if t == 1:
        let i = reader.readInt - 1
        sSeq[i] = 1 - sSeq[i]
    else:
        let (l, r) = (reader.readInt - 1, reader.readInt)
        var cumsum = newSeq[int](r - l + 1)
        cumsum[0] = 0
        for i, x in sSeq[l..<r]:
            cumsum[i + 1] = cumsum[i] + x
        
        var ans = 0
        for i in 0..<cumsum.len:
            for j in i+1..<cumsum.len:
                let len = j - i
                let x = cumsum[j] - cumsum[i]
                ans = (ans + x * (len - x)) mod MOD
        echo ans
0