結果

問題 No.2210 equence Squence Seuence
ユーザー Shirotsume
提出日時 2023-02-05 18:59:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,118 ms / 2,000 ms
コード長 2,054 bytes
コンパイル時間 203 ms
コンパイル使用メモリ 82,492 KB
実行使用メモリ 136,040 KB
最終ジャッジ日時 2024-07-04 06:30:50
合計ジャッジ時間 12,028 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import deque, Counter
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
inf = 2 ** 63 - 1
mod = 998244353
from functools import cmp_to_key
class Rollinghash:
    def __init__(self, string, base, mod):
        n = len(string)
        self.__base = base
        self.__mod = mod
        self.__hash = [0]*(n + 1)
        self.__pow = [1]*(n + 1)
        if isinstance(string, str):
            for i, c in enumerate(string):
                o = ord(c) - ord('a') + 1
                self.__hash[i + 1] = (self.__hash[i] * self.__base + o) % self.__mod
                self.__pow[i + 1] = self.__pow[i] * self.__base % self.__mod
        else:
            for i, c in enumerate(string):
                o = c
                self.__hash[i + 1] = (self.__hash[i] * self.__base + o) % self.__mod
                self.__pow[i + 1] = self.__pow[i] * self.__base % self.__mod
        self.pow = self.__pow


   
    def query(self, l, r):
        ret = (self.__hash[r] - self.__hash[l] * self.__pow[r - l]) % self.__mod
        return ret
    def same(self, l1, r1, l2, r2):
        return self.query(l1, r1) == self.query(l2, r2)
import random
n, k = mi()
a = li()
base = random.randint(1, 998244352)
S = Rollinghash(a, base, mod)

def calchash(i, r):
    if r < i:
        return S.query(0,r)
    else:
        return (S.query(0, i) * S.pow[r - i] + S.query(i + 1,r + 1)) % mod

def ele(i, x):
    if x < i:
        return a[x]
    else:
        return a[x + 1]
def cmp(x, y):
    if x == y:
        return 0
    if calchash(x, n - 1) == calchash(y, n - 1):
        return 0
    ng = n - 1
    ok = 0
    while abs(ok - ng) > 1:
        mid = (ok + ng) // 2
        if calchash(x, mid) == calchash(y, mid):
            ok = mid
        else:
            ng = mid
    
    if ele(x, ok) < ele(y, ok):
        return -1
    else:
        return 1

p = sorted(list(range(n)), key = cmp_to_key(cmp))
x = p[k - 1]

ans = a[:x] + a[x + 1:]

print(*ans)
0