結果

問題 No.1994 Confusing Name
ユーザー lam6er
提出日時 2025-03-20 21:04:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 307 ms / 2,000 ms
コード長 955 bytes
コンパイル時間 233 ms
コンパイル使用メモリ 82,392 KB
実行使用メモリ 134,824 KB
最終ジャッジ日時 2025-03-20 21:05:00
合計ジャッジ時間 6,756 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import defaultdict

def main():
    n = int(sys.stdin.readline())
    all_strings = [sys.stdin.readline().strip() for _ in range(n)]
    results = [0] * n

    len_groups = defaultdict(list)
    for idx, s in enumerate(all_strings):
        length = len(s)
        len_groups[length].append((idx, s))
    
    for length, group in len_groups.items():
        group_strings = [s for idx, s in group]
        mask_key_counts = defaultdict(int)
        for s in group_strings:
            for i in range(length):
                key = s[:i] + '*' + s[i+1:]
                mask_key_counts[key] += 1
        
        for original_idx, s in group:
            total = 0
            for i in range(length):
                key = s[:i] + '*' + s[i+1:]
                total += mask_key_counts[key] - 1
            results[original_idx] = total
    
    for count in results:
        print(count)

if __name__ == "__main__":
    main()
0