結果

問題 No.852 連続部分文字列
ユーザー stngstng
提出日時 2022-07-23 13:31:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,382 ms / 3,153 ms
コード長 1,354 bytes
コンパイル時間 161 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 158,684 KB
最終ジャッジ日時 2024-07-05 01:25:43
合計ジャッジ時間 19,416 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
51,968 KB
testcase_01 AC 40 ms
52,224 KB
testcase_02 AC 39 ms
52,608 KB
testcase_03 AC 38 ms
52,480 KB
testcase_04 AC 39 ms
52,352 KB
testcase_05 AC 44 ms
58,752 KB
testcase_06 AC 45 ms
58,368 KB
testcase_07 AC 42 ms
58,624 KB
testcase_08 AC 41 ms
57,472 KB
testcase_09 AC 40 ms
52,608 KB
testcase_10 AC 42 ms
58,496 KB
testcase_11 AC 41 ms
57,856 KB
testcase_12 AC 43 ms
58,368 KB
testcase_13 AC 102 ms
77,312 KB
testcase_14 AC 89 ms
77,212 KB
testcase_15 AC 120 ms
79,488 KB
testcase_16 AC 102 ms
77,312 KB
testcase_17 AC 111 ms
77,608 KB
testcase_18 AC 121 ms
78,428 KB
testcase_19 AC 83 ms
76,800 KB
testcase_20 AC 95 ms
76,800 KB
testcase_21 AC 120 ms
78,264 KB
testcase_22 AC 91 ms
77,184 KB
testcase_23 AC 105 ms
77,440 KB
testcase_24 AC 95 ms
76,928 KB
testcase_25 AC 107 ms
77,712 KB
testcase_26 AC 119 ms
78,144 KB
testcase_27 AC 102 ms
78,464 KB
testcase_28 AC 111 ms
77,988 KB
testcase_29 AC 122 ms
78,116 KB
testcase_30 AC 85 ms
76,660 KB
testcase_31 AC 122 ms
78,280 KB
testcase_32 AC 126 ms
78,592 KB
testcase_33 AC 1,347 ms
158,092 KB
testcase_34 AC 1,354 ms
158,172 KB
testcase_35 AC 1,382 ms
158,560 KB
testcase_36 AC 1,298 ms
158,684 KB
testcase_37 AC 1,303 ms
157,912 KB
testcase_38 AC 1,285 ms
158,384 KB
testcase_39 AC 1,298 ms
157,948 KB
testcase_40 AC 1,298 ms
158,000 KB
testcase_41 AC 1,290 ms
157,988 KB
testcase_42 AC 1,289 ms
154,128 KB
testcase_43 AC 1,310 ms
132,184 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Bit:
    def __init__(self, n):
        self.size = n
        self.tree = [0] * (n + 1)
 
    def sum(self, i):
        s = 0
        while i > 0:
            s += self.tree[i]
            i -= i & -i
        return s
 
    def add(self, i, x):
        while i <= self.size:
            self.tree[i] += x
            i += i & -i

class RangeUpdate:
    def __init__(self, n):
        self.p = Bit(n + 1)
        self.q = Bit(n + 1)
     
    def add(self, s, t, x):
        t += 1
        self.p.add(s, -x * s)
        self.p.add(t, x * t)
        self.q.add(s, x)
        self.q.add(t, -x)
     
    def sum(self, s, t):
        t += 1
        return self.p.sum(t) + self.q.sum(t) * t - \
               self.p.sum(s) - self.q.sum(s) * s
    #yukicoder218B

s = input()
ngs = len(s)

alp = [[] for i in range(26)]
li = [0]*(ngs)
rui = [0]*(ngs+1)

for i in range(ngs):
    idx = ord(s[i]) - ord("a")
    li[i] = li[i-1]
    if len(alp[idx]) == 0:
        li[i] += 1
    alp[idx].append(i)

ru = RangeUpdate(ngs+1)

for i in range(ngs):
    ru.add(i+1,i+1,li[i])

ans = ru.sum(1,ngs)
aidx = [1]*26
for i in range(1,ngs):
    idx = ord(s[i-1]) - ord("a")
    if len(alp[idx]) <= aidx[idx]:
        ru.add(i+1,ngs,-1)
    else:
        ru.add(i+1,alp[idx][aidx[idx]],-1)
        aidx[idx] += 1
    ans += ru.sum(i+1,ngs)
print(ans/((ngs*(ngs+1))/2))
0