AT_joi2026_yo1b_d 距離 (Distance)

题目描述

有 $N$ 个小球排成一列。从左到右依次编号为 $1,2,\dots,N$。第 $i$ 个小球被涂成颜色 $C_i$。 对于每个 $i=1,2,\dots,N$,求以下数值: - 所有与第 $i$ 个小球颜色相同的小球,与第 $i$ 个小球之间的距离之和。这里,定义第 $i$ 个小球与第 $j$ 个小球的距离为 $|i-j|$。

输入格式

输入格式如下: > $N$ $C_1$ $C_2$ $\dots$ $C_N$

输出格式

输出 $N$ 行。第 $i$ 行($1 \leq i \leq N$)输出所有与第 $i$ 个小球颜色相同的小球,与第 $i$ 个小球之间的距离总和。 除答案外,不得输出任何内容(例如,不要输出提示输入等无关文本)。 关于解答输出的格式,可以参考 [练习题及其参考答案](https://atcoder.jp/contests/joi2026-practice)。

说明/提示

### 样例解释 1 第 $1$ 个小球为颜色 $1$,第 $2$ 个小球为颜色 $2$,第 $3$ 个小球为颜色 $1$,第 $4$ 个小球为颜色 $1$。 - 对于 $i=1$,与第 $1$ 个小球颜色相同的小球有第 $1$、$3$、$4$ 个,其与第 $1$ 个小球的距离分别为 $0,2,3$,因此输出 $0+2+3=5$。 - 对于 $i=2$,与第 $2$ 个小球颜色相同的小球只有第 $2$ 个,与自己的距离为 $0$,因此输出 $0$。 - 对于 $i=3$,与第 $3$ 个小球颜色相同的小球有第 $1$、$3$、$4$ 个,其与第 $3$ 个小球的距离分别为 $2,0,1$,因此输出 $2+0+1=3$。 - 对于 $i=4$,与第 $4$ 个小球颜色相同的小球有第 $1$、$3$、$4$ 个,其与第 $4$ 个小球的距离分别为 $3,1,0$,因此输出 $3+1+0=4$。 ### 样例解释 2 第 $1$ 个小球为颜色 $100$。 - 对于 $i=1$,与第 $1$ 个小球颜色相同的小球只有第 $1$ 个,与自己的距离为 $0$,因此输出 $0$。 ### 数据范围 - $1 \leq N \leq 100$。 - $1 \leq C_i \leq 100$($1 \leq i \leq N$)。 - 输入的所有数均为整数。 由 ChatGPT 5 翻译