結果

問題 No.2020 Sum of Common Prefix Length
コンテスト
ユーザー rin204
提出日時 2022-07-22 22:48:10
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 3,206 bytes
コンパイル時間 197 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 111,216 KB
最終ジャッジ日時 2024-07-04 07:20:16
合計ジャッジ時間 13,210 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 15 TLE * 1 -- * 22
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import os
import sys
from io import BytesIO, IOBase
 
BUFSIZE = 8192
 
 
class FastIO(IOBase):
    newlines = 0
 
    def __init__(self, file):
        self._fd = file.fileno()
        self.buffer = BytesIO()
        self.writable = "x" in file.mode or "r" not in file.mode
        self.write = self.buffer.write if self.writable else None
 
    def read(self):
        while True:
            b = os.read(self._fd, max(os.fstat(self._fd).st_size, BUFSIZE))
            if not b:
                break
            ptr = self.buffer.tell()
            self.buffer.seek(0, 2), self.buffer.write(b), self.buffer.seek(ptr)
        self.newlines = 0
        return self.buffer.read()
 
    def readline(self):
        while self.newlines == 0:
            b = os.read(self._fd, max(os.fstat(self._fd).st_size, BUFSIZE))
            self.newlines = b.count(b"\n") + (not b)
            ptr = self.buffer.tell()
            self.buffer.seek(0, 2), self.buffer.write(b), self.buffer.seek(ptr)
        self.newlines -= 1
        return self.buffer.readline()
 
    def flush(self):
        if self.writable:
            os.write(self._fd, self.buffer.getvalue())
            self.buffer.truncate(0), self.buffer.seek(0)
 
 
class IOWrapper(IOBase):
    def __init__(self, file):
        self.buffer = FastIO(file)
        self.flush = self.buffer.flush
        self.writable = self.buffer.writable
        self.write = lambda s: self.buffer.write(s.encode("ascii"))
        self.read = lambda: self.buffer.read().decode("ascii")
        self.readline = lambda: self.buffer.readline().decode("ascii")
 
 
sys.stdin, sys.stdout = IOWrapper(sys.stdin), IOWrapper(sys.stdout)
input = lambda: sys.stdin.readline().rstrip("\r\n")

class Node:
    def __init__(self):
        self.nex = [None] * 26
        self.ind = []
        self.tot = 0

n = int(input())
V0 = Node()
Vs = [None] * n
ans = [0] * n
SS = []
for i in range(n):
    S = input()
    SS.append([ord(s) - 97 for s in S])
    V = V0
    for s in SS[i]:
        if V.nex[s] is None:
            V.nex[s] = Node()
        V = V.nex[s]
        V.tot += 1
    Vs[i] = V

B = 750
for i, S in enumerate(SS):
    V = V0
    flg = len(S) >= B
    for s in S:
        V = V.nex[s]
        ans[i] += V.tot
        if flg:
            V.ind.append(i)

Q = int(input())
for _ in range(Q):
    query = input().split()
    if query[0] == "1":
        i = int(query[1]) - 1
        s = query[2]
        s = ord(s) - 97
        SS[i].append(s)
        V = Vs[i]
        if V.nex[s] is None:
            V.nex[s] = Node()
        V = V.nex[s]
        V.tot += 1
        for j in V.ind:
            ans[j] += 1
        le = len(SS)
        if le < B:
            pass
        elif le == B:
            ans[i] = 0
            V = V0
            for s in SS[i]:
                V = V.nex[s]
                ans[i] += V.tot
                V.ind.append(i)
        else:
            V.ind.append(i)
            ans[i] += V.tot
        Vs[i] = V

    else:
        i = int(query[1]) - 1
        le = len(SS[i])
        if le < B:
            ans[i] = 0
            V = V0
            for s in SS[i]:
                V = V.nex[s]
                ans[i] += V.tot
        print(ans[i])
0