結果
問題 | 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 = n self.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 = 0 while i>0: s += self.tree[i] i -= i & -i return s # 0 index を 1 index に変更 転倒数を求めるなら1を足していく def add(self, i, x): i += 1 while i <= self.size: self.tree[i] += x i += i & -i # 総和がx以上になる位置のindex をbinary search def bsearch(self,x): le = 0 ri = 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 += ri ri >>= 1 return le+1 n,m = map(int,input().split()) P = [int(x)-1 for x in input().split()] num = 0 bit = 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%m if dif%2: dif += m print(num+dif)