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 翻译