提出 #68284819

---

ソースコード 拡げる

# Python 3.11.4
# 処理時間: O(N)
# メモリ使用量: O(U) (U は配列内の異なる値の数、最大で N)

from sys import stdin
from typing import List


def count_pairs(N: int, arr: List[int]) -> int:
    """
    条件を満たす (i, j) の組数をカウントする

    Parameters
    ----------
    N : int
        配列の要素数
    arr : List[int]
        長さ N の整数配列 A1...AN

    Returns
    -------
    int
        条件 (1 ≤ j < i ≤ N かつ A_j = A_i) を満たす組数
    """
    freq: dict[int, int] = {}  # 値 -> 出現回数
    count: int = 0

    for val in arr:
        if val in freq:
            # 過去に出現した回数分だけ組を作れる
            count += freq[val]
            freq[val] += 1
        else:
            freq[val] = 1
    return count


# 高速な入力処理
data: List[int] = list(map(int, stdin.read().strip().split()))
N: int = data[0]
arr: List[int] = data[1:]

result: int = count_pairs(N, arr)
print(result)

提出情報

ジャッジ結果