019 - Choose Cards 1 Editorial /

Time Limit: 1 sec / Memory Limit: 1024 MiB

配点: 1000

問題文

N 枚のカードがあり、左から i 番目(1 \leq i \leq N)のカードの色は A_i です。 A_i=1 のとき赤色、A_i=2 のとき黄色、A_i=3 のとき青色です。同じ色のカードを 2 枚選ぶ方法は何通りありますか。

制約

  • 2 \leq N \leq 500000
  • 1 \leq A_i \leq 3
  • 入力はすべて整数

入力

入力は以下の形式で標準入力から与えられます。

N
A_1 A_2 \cdots A_N

出力

答えを出力してください。


入力例 1

6
1 3 2 1 1 2

出力例 1

4

以下の 4 通りの方法があります。

  • 左から 1 番目のカードと、左から 4 番目のカードを選ぶ。
  • 左から 1 番目のカードと、左から 5 番目のカードを選ぶ。
  • 左から 4 番目のカードと、左から 5 番目のカードを選ぶ。
  • 左から 3 番目のカードと、左から 6 番目のカードを選ぶ。