1. Job Roadmap
  2. Home
  3. Problem Set
  4. codenotelist
  5. Forum
  6. course
  7. Shore Share Sessions
  8. Record
  1. Login
  2. Sign Up
  3. Language
    1. English
    2. 한국어
    3. 简体中文
    4. 正體中文
    ZhContent TextSol AI分析

解题思路

给定一棵以 1 为根的树,每个节点有权值 aia_iai​。对每个节点 iii,要求统计在从根 1 到该节点 iii 的简单路径上,有多少无序对 (x,y)(x,y)(x,y)(x≠yx\ne yx=y)满足 ax=aya_x=a_yax​=ay​。

核心做法:根到当前节点的前缀统计 + DFS(栈模拟)

  • 沿着树做一次从根出发的 DFS。维护一张“当前路径上各权值出现次数”的表 freq,以及“当前路径上的相等权值对总数” pairs。
  • 当进入一个节点 uuu 时(把它加入根到当前的路径):

P3769.第1题-节点树的简单路径

    1000ms Tried: 111 Accepted: 43 Difficulty: 5 所属公司 : 字节
    算法与标签>DFS

题目内容

Bingbong 有一棵 nnn 个节点的树,编号为 111~ nnn ,节点 iii 的权值为 aia_iai​ 。

现在你需要计算对于任意的节点 i∈∣1,n∣i∈|1,n|i∈∣1,n∣ ,有多少对 (x,y)(x≠y)(x,y)(x≠y)(x,y)(x=y) ,满足 x,yx,yx,y 两个节点均在 1→i1→i1→i 的简单路径上,且 ax=aya_x= a_yax​=ay​ 。

输入描述

第一行一个数 n(2≦n≦2×105)n(2≦n≦2×10^5)n(2≦n≦2×105) ,表示节点总数。

第二行 nnn 个整数,表示节点 iii 的权值 ai(1≦n≦109)a_i(1≦n≦10^9)ai​(1≦n≦109) 。

接下来 n−1n-1n−1 行,每行 222 个整数 u,v(1≦u,v≦n)u,v(1≦u,v≦n)u,v(1≦u,v≦n) ,表示当前无向边连接 u,vu,vu,v 两个节点。

保证输入是一棵树。

输出描述

输出包含一行,共 nnn 个整数,每个整数之同以空格隔开,含义如题所示。

样例1

输入

4
1 1 2 2
1 2
2 3
3 4

输出

0 1 1 2

登录后即可使用 AI 分析。

模式
倒计时时长
:

最长 10 小时 59 分;应用后按此时长重新开始。

提示:点击提交记录在左侧题面区域查看详情
题库
AI分析设置
留空使用官方API Key,每天有次数限制(自定义API Key仅限会员和管理员使用,不限次数)
会员和管理员可切换模型;切到 Kimi/智谱/通义/豆包时需填写对应供应商 API Key
升级会员,可将运行与提交冷却时间缩短至 1 秒起

Status

  • Judging Queue
  • Service Status

Development

  • Open Source

Support

  • Help
  • Contact Us

About

  • About
  • Privacy
  • Terms of Service
  • Copyright Complaint
  1. Language
    1. English
    2. 한국어
    3. 简体中文
    4. 正體中文
  2. Legacy mode
  3. Theme
    1. Light
    2. Dark
  1. 京ICP备2025123107号-1
  2. Worker 2, 34ms
  3. Powered by Hydro v5.0.0-beta.18 Community
CLOSE


ScanQRCodePrompt

请使用微信扫描下方二维码完成注册

Forgot password or username?