#820515 (PyPy3) No.2170 Left Addition Machine

• 問題: No.2170 Left Addition Machine
• ユーザー: 👑 SPD_9X2
• 提出日時: 2022-11-28 17:01:56
• 言語: PyPy3 (7.3.15)
• 結果: TLE
• 実行時間: -
• コード長: 2,330 bytes
• コンパイル時間: 424 ms
• コンパイル使用メモリ: 82,320 KB
• 実行使用メモリ: 303,900 KB
• 最終ジャッジ日時: 2024-11-18 06:01:52
• 合計ジャッジ時間: 43,607 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

ファイルパターン	結果
other	AC * 62 TLE * 7

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
"""
Left Addition Machine 想定TLE解 2
間に、A[i-1] >= A[i] となるi-1,iが存在する場合だけ線形
"""
import sys
from sys import stdin
def rangesum(arr,l,r): #[l:r] (0-indexed) の和を取る arrはあらかじめ累積和しておくこと
    if l == 0:
        return arr[r]
    else:
        return arr[r] - arr[l-1]
    
mod = 998244353
#入力を受け取る
N,Q = map(int,stdin.readline().split())
A = list(map(int,stdin.readline().split()))
LR = []
for i in range(Q):
    L,R = map(int,stdin.readline().split())
    LR.append( (L,R) )
#制約チェック
assert 1 <= N <= 2*(10**5)
assert 1 <= Q <= 2*(10**5)
assert len(A) == N
assert len(LR) == Q
for i in range(N):
    assert 0 <= A[i] < mod
for i in range(Q):
    L,R = LR[i]
    assert 1 <= L <= R <= N
#A[0:r) までの場合を計算
fst  = [] #fst[i] = A[0:r) までを入れた場合の結果
fnew = [0] * N #A[i-1] >= A[i] ならば fnew[i] = 1とする。
for i in range(N):
    if i == 0 or A[i-1] >= A[i]:
        fst.append(A[i])
        fnew[i] = 1
        pw = 1
    else:
        fst.append( (fst[-1] + pw * A[i]) % mod )
        pw *= 2
        pw %= mod
#fnewの累積和を取っておく
for i in range(N-1):
    fnew[i+1] += fnew[i]
powtwo = [1] #2^i を前計算
invtwo = [1] #1/2^i を前計算
half = pow(2,mod-2,mod) #1/2 (mod 998244353)
for i in range(N):
    powtwo.append(powtwo[-1] * 2 % mod)
    invtwo.append(invtwo[-1] * half % mod)
CS = [A[i] * powtwo[i] % mod for i in range(N)] #解説の配列Cの累積和
for i in range(N-1): #累積和を取る
    CS[i+1] += CS[i]
ANS = []
for loop in range(Q): #クエリを処理
    l,r = LR[loop]
    l -= 1 #0-indexedにする
    r -= 1
 
    if fnew[l] != fnew[r]: #間に、A[i-1] >= A[i] となるi-1,iが存在する
        #ANS.append( fst[r] % mod )
        """
        変更箇所はこの辺。線形になっている。
        """
        ans = 0
        for i in range(r,l-1,-1):
            if A[i-1] >= A[i]:
                ans += A[i]
                ans %= mod
                break
            else:
                ans *= 2
                ans += A[i]
                ans %= mod
        ANS.append(ans)
            
    else:
        ANS.append( (A[l] + rangesum(CS,l+1,r) * invtwo[l+1]) % mod )
print ("\n".join(map(str,ANS)))
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX