結果
問題 |
No.696 square1001 and Permutation 5
|
ユーザー |
|
提出日時 | 2023-06-13 19:52:39 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,540 bytes |
コンパイル時間 | 1,961 ms |
コンパイル使用メモリ | 81,996 KB |
実行使用メモリ | 93,356 KB |
最終ジャッジ日時 | 2024-06-22 05:47:11 |
合計ジャッジ時間 | 3,041 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 9 RE * 3 |
ソースコード
import sys sys.set_int_max_str_digits(150000) N = int(input()) P = tuple(map(int, input().split())) class BIT(): def __init__(self, sequence): self.size = len(sequence) self.depth = self.size.bit_length() self.build(sequence) def build(self, sequence): data = sequence size = self.size for I, X in enumerate(data): J = I + (I & (-I)) if J < size: data[J] += data[I] self.data = data def SUM(self, I): data = self.data S = 0 while I: S += data[I] I -= I & -I return S def ADD(self, I, X): data = self.data size = self.size while I < size: data[I] += X I += I & -I def FIND(self, K): data = self.data size = self.size X, SX = 0, 0 DX = 1 << (self.depth) for i in range(self.depth - 1, -1, -1): DX = (1 << i) if X + DX >= size: continue Y = X + DX SY = SX + data[Y] if SY < K: X, SX = Y, SY return X + 1 bit, g = BIT([0] * (N + 1)), [0] * N for I, X in enumerate(P): g[I] = X - 1 - bit.SUM(X) bit.ADD(X, 1) NUMS, COES = g[::-1], list(range(1, N + 1)) while True: N = len(NUMS) if N == 1: break for i in range(1, N, 2): NUMS[i - 1] += COES[i - 1] * NUMS[i] COES[i - 1] *= COES[i] NUMS = NUMS[::2] COES = COES[::2] print(NUMS[0] + 1)