結果
問題 | No.2419 MMA文字列2 |
ユーザー |
|
提出日時 | 2023-08-12 14:46:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 511 ms / 2,000 ms |
コード長 | 974 bytes |
コンパイル時間 | 307 ms |
コンパイル使用メモリ | 82,664 KB |
実行使用メモリ | 256,336 KB |
最終ジャッジ日時 | 2024-11-19 21:23:28 |
合計ジャッジ時間 | 8,348 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
from math import comb s = input() # i文字目以降にあるアルファベットが何回出現するかを調べる ss = [[0] for _ in range(26)] for si in s: for i in range(26): si_int = ord(si) - ord("A") if si_int == i: ss[i].append(ss[i][-1] + 1) else: ss[i].append(ss[i][-1]) ans = 0 for i, si in enumerate(s): si = ord(si) - ord("A") # i文字目以降にあるS_i以外のアルファベットを数えたい # 最初二文字の選択肢がいくつ増える?普通に組み合わせ計算して引こうぜ # i文字目までにS_iが何個あるか? if ss[si][i + 1] == 2: # i文字目以降のS_i以外のアルファベットの個数 ans += (len(s) - i - 1) - (ss[si][-1] - ss[si][i + 1]) elif ss[si][i + 1] > 2: ans += (comb(ss[si][i + 1], 2) - comb(ss[si][i], 2)) * ( (len(s) - i - 1) - (ss[si][-1] - ss[si][i + 1]) ) print(ans)