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 番目のカードを選ぶ。