結果
問題 | No.1975 Zigzag Sequence |
ユーザー |
![]() |
提出日時 | 2023-02-23 07:57:41 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 753 ms / 2,000 ms |
コード長 | 1,876 bytes |
コンパイル時間 | 288 ms |
コンパイル使用メモリ | 82,412 KB |
実行使用メモリ | 108,372 KB |
最終ジャッジ日時 | 2024-07-23 05:47:25 |
合計ジャッジ時間 | 15,354 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
import sys# sys.setrecursionlimit(200005)int1 = lambda x: int(x)-1pDB = lambda *x: print(*x, end="\n", file=sys.stderr)p2D = lambda x: print(*x, sep="\n", end="\n\n", file=sys.stderr)def II(): return int(sys.stdin.readline())def LI(): return list(map(int, sys.stdin.readline().split()))def LLI(rows_number): return [LI() for _ in range(rows_number)]def LI1(): return list(map(int1, sys.stdin.readline().split()))def LLI1(rows_number): return [LI1() for _ in range(rows_number)]def SI(): return sys.stdin.readline().rstrip()# dij = [(0, 1), (-1, 0), (0, -1), (1, 0)]dij = [(0, 1), (-1, 0), (0, -1), (1, 0), (1, 1), (1, -1), (-1, 1), (-1, -1)]inf = (1 << 63)-1# inf = (1 << 31)-1md = 10**9+7# md = 998244353class BitSum:def __init__(self, n):self.n = n+1self.table = [0]*self.nself.aa = [0]*ndef __getitem__(self, item):return self.aa[item]def add(self, i, x):self.aa[i] += xi += 1while i < self.n:self.table[i] += xi += i & -idef sum(self, i):i += 1res = 0while i > 0:res += self.table[i]i -= i & -ireturn resdef sumlr(self, l, r):if l >= r: return 0if l == 0: return self.sum(r-1)return self.sum(r-1)-self.sum(l-1)def solve():res = 0ia = sorted(enumerate(aa), key=lambda x: x[1])ll, rr = BitSum(n), BitSum(n)pend = []for i, a in ia:if pend and pend[0][1] != a:for j, _ in pend:ll.add(j, pow(2, j, md))j = n-1-jrr.add(j, pow(2, j, md))pend.clear()res += ll.sum(i)*rr.sum(n-1-i)%mdres %= mdpend.append((i, a))return resn = II()aa = LI()ans = solve()aa = [-a for a in aa]ans += solve()print(ans%md)