最新公告
  • 欢迎您光临信息学奥赛网,一个优质的信息学编程题库和信息学编程学习资源专业网站。欢迎加入VIP
  • 树上游戏

    正文概述 陈老师   2026-01-20 15:15:23  

    题目描述

    lrb 有一棵树,树的每个节点有个颜色。给一个长度为 nn 的颜色序列,定义 s(i,j)s(i,j)iijj 的颜色数量。以及

    sumi=j=1ns(i,j)sum_i=\sum_{j=1}^n s(i, j)

    现在他想让你求出所有的 sumisum_i

    输入格式

    第一行为一个整数 nn,表示树节点的数量。

    第二行为 nn 个整数,分别表示 nn 个节点的颜色 c1,c2cnc_1,c_2\ldots c_n

    接下来 n1n-1 行,每行为两个整数 x,yx,y,表示 xxyy 之间有一条边。

    输出格式

    输出 nn 行,第 ii 行为 sumisum_i

    样例

    5
    1 2 3 2 3
    1 2
    2 3
    2 4
    1 5
    
    10
    9
    11
    9
    12
    
    $$sum_1=s(1,1)+s(1,2)+s(1,3)+s(1,4)+s(1,5)=1+2+3+2+2=10$$$$sum_2=s(2,1)+s(2,2)+s(2,3)+s(2,4)+s(2,5)=2+1+2+1+3=9$$$$sum_3=s(3,1)+s(3,2)+s(3,3)+s(3,4)+s(3,5)=3+2+1+2+3=11$$$$sum_4=s(4,1)+s(4,2)+s(4,3)+s(4,4)+s(4,5)=2+1+2+1+3=9$$$$sum_5=s(5,1)+s(5,2)+s(5,3)+s(5,4)+s(5,5)=2+3+3+3+1=12$$

    数据范围与约定

    对于 40%40\% 的数据,n2000n\leq 2000

    对于 100%100\% 的数据,1n,ci1051\leq n,c_i\leq 10^5

    信息学奥赛网,一个优质的信息学奥赛学习资源平台!
    信息学奥赛网 » 树上游戏