题意:一棵有n个点的树,树上每个点都有颜色c[i],定义每条路径的值为这条路径上经过的不同颜色数量和。求所有路径的值的和。

可以把问题转化为对每种颜色有多少条不同的路径至少经过这种颜色的点,然后加和。求有多少条路径经过可以转换为总路径数-没有经过的路径数,只要求出没有经过的路径数就好了。

对于每一个相同颜色的点,它们将树割成一些个联通块,显然这些联通块内部之间的路径不会经过这种颜色。

于是问题转化为求点划分的联通块大小。

用类似于虚树的dfs办法,每次维护树上最左边的一段链,然后用栈进行数据的更新即可。

  1. # include <cstdio>
  2. # include <cstring>
  3. # include <cstdlib>
  4. # include <iostream>
  5. # include <vector>
  6. # include <queue>
  7. # include <stack>
  8. # include <map>
  9. # include <bitset>
  10. # include <set>
  11. # include <cmath>
  12. # include <algorithm>
  13. using namespace std;
  14. # define lowbit(x) ((x)&(-x))
  15. # define pi acos(-1.0)
  16. # define eps 1e-
  17. # define MOD
  18. # define INF
  19. # define mem(a,b) memset(a,b,sizeof(a))
  20. # define FOR(i,a,n) for(int i=a; i<=n; ++i)
  21. # define FDR(i,a,n) for(int i=a; i>=n; --i)
  22. # define bug puts("H");
  23. # define lch p<<,l,mid
  24. # define rch p<<|,mid+,r
  25. # define mp make_pair
  26. # define pb push_back
  27. typedef pair<int,int> PII;
  28. typedef vector<int> VI;
  29. # pragma comment(linker, "/STACK:1024000000,1024000000")
  30. typedef long long LL;
  31. inline int Scan() {
  32. int x=,f=; char ch=getchar();
  33. while(ch<''||ch>''){if(ch=='-') f=-; ch=getchar();}
  34. while(ch>=''&&ch<=''){x=x*+ch-''; ch=getchar();}
  35. return x*f;
  36. }
  37. inline void Out(int a) {
  38. if(a<) {putchar('-'); a=-a;}
  39. if(a>=) Out(a/);
  40. putchar(a%+'');
  41. }
  42. const int N=;
  43. //Code begin...
  44.  
  45. struct Edge{int p, next;}edge[N<<];
  46. int head[N], cnt=;
  47. int node[N], siz[N], num[N], tmp, sum[N], mark;
  48. int st[N<<], f[N], col[N], pos;
  49. LL ans=;
  50.  
  51. void add_edge(int u, int v){edge[cnt].p=v; edge[cnt].next=head[u]; head[u]=cnt++;}
  52. void dfs(int x, int fa){
  53. siz[x]=;
  54. for (int i=head[x]; i; i=edge[i].next) {
  55. int v=edge[i].p;
  56. if (v==fa) continue;
  57. dfs(v,x); siz[x]+=siz[v];
  58. }
  59. }
  60. void sol(int x, int fa){
  61. num[x]=siz[x]; --sum[node[x]];
  62. if (node[x]==node[fa]) --num[x];
  63. if (col[node[x]]) tmp=st[col[node[x]]], --num[tmp];
  64. if (fa) {
  65. if (col[node[fa]]) tmp=st[col[node[fa]]], num[tmp]-=num[x];
  66. f[++pos]=col[node[fa]]; col[node[fa]]=pos; st[pos]=x;
  67. }
  68. for (int i=head[x]; i; i=edge[i].next) {
  69. int v=edge[i].p;
  70. if (v==fa) continue;
  71. sol(v,x);
  72. }
  73. if (fa) {
  74. ans+=(LL)num[x]*(num[x]-)/; sum[node[fa]]-=num[x];
  75. col[node[fa]]=f[col[node[fa]]];
  76. }
  77. }
  78. void init(){
  79. mem(head,); mem(siz,); mem(num,); mem(sum,); mem(col,); mem(f,);
  80. ans=; mark=; cnt=; pos=;
  81. }
  82. int main ()
  83. {
  84. int cas=, n, u, v;
  85. while (~scanf("%d",&n)) {
  86. init();
  87. FOR(i,,n) scanf("%d",node+i), sum[node[i]]=n;
  88. FOR(i,,n) if (sum[i]) ++mark;
  89. FOR(i,,n-) scanf("%d%d",&u,&v), add_edge(u,v), add_edge(v,u);
  90. dfs(,);
  91. sol(,);
  92. FOR(i,,n) ans+=(LL)sum[i]*(sum[i]-)/;
  93. ans=(LL)mark*n*(n-)/-ans;
  94. printf("Case #%d: %lld\n",++cas,ans);
  95. }
  96. return ;
  97. }

HDU 6035 Colorful Tree(dfs)的更多相关文章

  1. HDU 6035 - Colorful Tree | 2017 Multi-University Training Contest 1

    /* HDU 6035 - Colorful Tree [ DFS,分块 ] 题意: n个节点的树,每个节点有一种颜色(1~n),一条路径的权值是这条路上不同的颜色的数量,问所有路径(n*(n-1)/ ...

  2. 2017 Multi-University Training Contest - Team 1 1003&&HDU 6035 Colorful Tree【树形dp】

    Colorful Tree Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 131072/131072 K (Java/Others)T ...

  3. 2017ACM暑期多校联合训练 - Team 1 1003 HDU 6035 Colorful Tree (dfs)

    题目链接 Problem Description There is a tree with n nodes, each of which has a type of color represented ...

  4. HDU 6035 Colorful Tree(补集思想+树形DP)

    [题目链接] http://acm.hdu.edu.cn/showproblem.php?pid=6035 [题目大意] 给出一颗树,一条路径的价值为其上点权的种类数,求路径总价值 [题解] 单独考虑 ...

  5. HDU 6035 Colorful Tree (树形DP)

    [题目链接] http://acm.hdu.edu.cn/showproblem.php?pid=6035 [题目大意] 给出一颗树,一条路径的价值为其上点权的种类数,求路径总价值 [题解] 我们计算 ...

  6. hdu 6035:Colorful Tree (2017 多校第一场 1003) 【树形dp】

    题目链接 单独考虑每一种颜色,答案就是对于每种颜色至少经过一次这种的路径条数之和.反过来思考只需要求有多少条路径没有经过这种颜色即可. 具体实现过程比较复杂,很神奇的一个树形dp,下面给出一个含较详细 ...

  7. hdu 6035 Colorful Tree(虚树)

    考虑到树上操作:首先题目要我们求每条路径上出现不同颜色的数量,并把所有加起来得到答案:我们知道俩俩点之间会形成一条路径,所以我们可以知道每个样例的总的路径的数目为:n*(n-1)/2: 这样单单的求, ...

  8. Hdu 5379 Mahjong tree (dfs + 组合数)

    题目链接: Hdu 5379 Mahjong tree 题目描述: 给出一个有n个节点的树,以节点1为根节点.问在满足兄弟节点连续 以及 子树包含节点连续 的条件下,有多少种编号方案给树上的n个点编号 ...

  9. 【hdu6035】 Colorful Tree dfs序

    题目传送门:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题目大意:给你一棵树,树上每个节点都有一个颜色. 现在定义两点间的距离为两点最短路径上颜色集合 ...

随机推荐

  1. 20155308 2016-2017-2《Java程序设计》课堂实践项目

    20155308 2016-2017-2<Java程序设计>课堂实践项目 在java.lang包中有String.split()方法,返回是一个数组 我在应用中用到一些,给大家总结一下,仅 ...

  2. linux 初学体验 20155317 王新玮

    学习基于VirtualBox虚拟机安装Ubuntu图文教程在自己笔记本上安装Linux操作系统 通过询问同学和上网搜查百度,完成了虚拟机的安装,开始了linux的学习. 通过实践学习别出心裁的Linu ...

  3. 成都优步uber司机奖励政策(持续更新)

    滴快车单单2.5倍,注册地址:http://www.udache.com/ 如何注册Uber司机(全国版最新最详细注册流程)/月入2万/不用抢单:http://didi-uber.com/archiv ...

  4. 《Node.js 包教不包会》

    <Node.js 包教不包会> 为何写作此课程 在 CNode(https://cnodejs.org/) 混了那么久,解答了不少 Node.js 初学者们的问题.回头想想,那些问题所需要 ...

  5. Entity Framework for Oracle 基本配置

    1.需要安装ODAC 如果不安装ODAC,在数据源连接的配置中,看不到Oracle的选项 我下载安装的组件是32-bit Oracle Data Access Components (ODAC)  w ...

  6. PHP基础知识试题

    转载于:http://www.php.cn/toutiao-415599.html 1.PHP中传值与传引用的区别,什么时候传值,什么时候传引用? 按值传递:函数范围内对值任何改变在函数外部都会被忽略 ...

  7. JS 中屏幕、浏览器和文档的高度、宽度和距离

    1.各种对象 window.screen - 屏幕,window - 窗口,document.documentElement & document.body.parentNode - 文档,d ...

  8. JUC——原子类操作(三)

    原子类操作 既然强调了并发访问,那么就必须考虑操作系统位数:32位操作系统还是64位操作系统,对于long型数据类型而言,是64位的.但是如果现在项目运行在32位系统上,则long型数据会占用32位空 ...

  9. 百度云 win10 125%界面模糊 解决

    右击图标 ->兼容性->更改高DPI设置 -> 替代高DPI缩放行为.打√

  10. HDU-1053:Advanced Fruits(LCS+路径保存)

    链接:HDU-1053:Advanced Fruits 题意:将两个字符串合成一个串,不改变原串的相对顺序,可将相同字母合成一个,求合成后最短的字符串. 题解:LCS有三种状态转移方式,将每个点的状态 ...