結果
問題 | No.2382 Amidakuji M |
ユーザー |
|
提出日時 | 2023-07-14 23:00:40 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 161 ms / 2,000 ms |
コード長 | 1,316 bytes |
コンパイル時間 | 442 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 95,872 KB |
最終ジャッジ日時 | 2024-09-16 08:08:40 |
合計ジャッジ時間 | 3,150 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
class BIT:def __init__(self, n):self.size = nself.tree = [0]*(n+1)def build(self, list):self.tree[1:] = list.copy()for i in range(self.size+1):j = i + (i & (-i))if j < self.size+1:self.tree[j] += self.tree[i]def sum(self, i):# [0, i) の要素の総和を返すs = 0while i>0:s += self.tree[i]i -= i & -ireturn s# 0 index を 1 index に変更 転倒数を求めるなら1を足していくdef add(self, i, x):i += 1while i <= self.size:self.tree[i] += xi += i & -i# 総和がx以上になる位置のindex をbinary searchdef bsearch(self,x):le = 0ri = 1<<(self.size.bit_length()-1)while ri > 0:if le+ri <= self.size and self.tree[le+ri]<x:x -= self.tree[le+ri]le += riri >>= 1return le+1n,m = map(int,input().split())P = [int(x)-1 for x in input().split()]num = 0bit = BIT(n+1)for i,p in enumerate(P):num += i-bit.sum(p)bit.add(p,1)if num%m == 0:print(num)elif num%2 != m%2 and num%2:print(-1)else:dif = m - num%mif dif%2:dif += mprint(num+dif)