[图论] 树上不重复权值的路径数

解题思路

整体思路,枚举路径上每个点,记录这个点 u 可以向上延伸到多远,如果可延伸到 x 点,则这个点对答案的贡献为 dep[u] - dep[x](下文的maxx) + 1

为什么枚举是向上延伸?因为可以记录已走过路径中已有的权值,每到一个新的点,只需要知道这个点在已有路径上是否存在。用来实现的数据结构是 map 以权值为键,以出现的深度为值

如果遇到已有权值,应进行的操作是:更新这个路径上能向上延伸的最大深度 maxx ,如果有已出现的权值点 v ,需要 maxx = dep[v]+1,所以最后这个点对答案的贡献为 ans += dep[u] - maxx

注意:当遍历结束一个结点,并要回溯到父结点时,需要把当前结点从map中弹出。

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define ll long long

const int N = 1e5+5;
vector<int> e[N];
int a[N];
map<int,vector<int>> m;
ll ans=0;

void dfs(int t,int dep,int maxx){
  if(m[a[t]].size()){
    maxx = max(m[a[t]].back(),maxx);
  }
  ans += dep-maxx;
  m[a[t]].push_back(dep);
  for(auto &y:e[t]){
    dfs(y,dep+1,maxx);
  }
  m[a[t]].pop_back();
  return ;
}


int main()
{
  int n;
  cin >> n;
  for(int i = 2 ; i <= n ; i++){
    int x;
    cin >> x;
    e[x].push_back(i);
  }
  for(int i = 1 ; i<=n ; i++){
    cin >> a[i];
  }
  dfs(1,1,0);
  cout << ans ;

  return 0;
}
相关推荐
127_127_1272 天前
2025 FJCPC 复建 VP
数据结构·图论·模拟·ad-hoc·分治·转化
wwer1425263632 天前
数学建模_图论
数学建模·图论
ysa0510303 天前
Dijkstra 算法#图论
数据结构·算法·图论
一只鱼^_3 天前
基础算法合集-图论
数据结构·算法·深度优先·图论·广度优先·宽度优先·图搜索算法
闻缺陷则喜何志丹15 天前
【并集查找】P10729 [NOISG 2023 Qualification] Dolls|普及+
c++·算法·图论·洛谷·并集查找
CodeWithMe15 天前
【Algorithm】图论入门
c++·图论
东方芷兰17 天前
Leetcode 刷题记录 13 —— 图论
算法·leetcode·图论
蒙奇D索大19 天前
【数据结构】图论实战:DAG空间压缩术——42%存储优化实战解析
数据结构·笔记·学习·考研·图论·改行学it
蒙奇D索大22 天前
【数据结构】图论最短路圣器:Floyd算法如何用双矩阵征服负权图?
数据结构·算法·矩阵·图论·图搜索算法
芜湖xin23 天前
【题解-洛谷】B4292 [蓝桥杯青少年组省赛 2022] 路线
算法·图论·bfs·图的遍历