AT_joi2026_yo1b_d.距離 (Distance)

入门

通过率:0%

AC君温馨提醒

该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

有 NN 个小球排成一列。从左到右依次编号为 1,2,…,N1,2,\dots,N。第 ii 个小球被涂成颜色 CiC_i。

对于每个 i=1,2,…,Ni=1,2,\dots,N,求以下数值:

  • 所有与第 ii 个小球颜色相同的小球,与第 ii 个小球之间的距离之和。这里,定义第 ii 个小球与第 jj 个小球的距离为 ∣i−j∣|i-j|。

输入格式

输入格式如下:

NN C1C_1 C2C_2 …\dots CNC_N

输出格式

输出 NN 行。第 ii 行(1≤i≤N1 \leq i \leq N)输出所有与第 ii 个小球颜色相同的小球,与第 ii 个小球之间的距离总和。

除答案外,不得输出任何内容(例如,不要输出提示输入等无关文本)。

关于解答输出的格式,可以参考 练习题及其参考答案。

输入输出样例

  • 输入#1

    4
    1 2 1 1

    输出#1

    5
    0
    3
    4
  • 输入#2

    1
    100

    输出#2

    0
  • 输入#3

    8
    1 3 9 7 3 3 7 9

    输出#3

    0
    7
    5
    3
    4
    5
    3
    5
  • 输入#4

    10
    1 1 1 1 1 1 1 1 1 1

    输出#4

    45
    37
    31
    27
    25
    25
    27
    31
    37
    45

说明/提示

样例解释 1

第 11 个小球为颜色 11,第 22 个小球为颜色 22,第 33 个小球为颜色 11,第 44 个小球为颜色 11。

  • 对于 i=1i=1,与第 11 个小球颜色相同的小球有第 11、33、44 个,其与第 11 个小球的距离分别为 0,2,30,2,3,因此输出 0+2+3=50+2+3=5。
  • 对于 i=2i=2,与第 22 个小球颜色相同的小球只有第 22 个,与自己的距离为 00,因此输出 00。
  • 对于 i=3i=3,与第 33 个小球颜色相同的小球有第 11、33、44 个,其与第 33 个小球的距离分别为 2,0,12,0,1,因此输出 2+0+1=32+0+1=3。
  • 对于 i=4i=4,与第 44 个小球颜色相同的小球有第 11、33、44 个,其与第 44 个小球的距离分别为 3,1,03,1,0,因此输出 3+1+0=43+1+0=4。

样例解释 2

第 11 个小球为颜色 100100。

  • 对于 i=1i=1,与第 11 个小球颜色相同的小球只有第 11 个,与自己的距离为 00,因此输出 00。

数据范围

  • 1≤N≤1001 \leq N \leq 100。
  • 1≤Ci≤1001 \leq C_i \leq 100(1≤i≤N1 \leq i \leq N)。
  • 输入的所有数均为整数。

由 ChatGPT 5 翻译

输入解题思路,AI测评打分。不知道怎么写?

首页