【题目链接】 http://acm.hdu.edu.cn/showproblem.php?pid=6035

【题目大意】

  给出一颗树,一条路径的价值为其上点权的种类数,求路径总价值

【题解】

  我们计算单个颜色的贡献,那么就是经过该颜色至少一次的路径数量,
  我们统计的时候在每个点记录以其为开始的路径的答案和,
  统计的时候计算了点自身,同时有重复计算的部分,最后减去n除以2即可
  那么我们只要在每种颜色的虚树上统计即可。
  对于子树的贡献需要区间修改,我们在dfs序的差分数组上更改,最后求前缀和即可。  

【代码】

  1. #include <cstdio>
  2. #include <algorithm>
  3. #include <cstring>
  4. #include <vector>
  5. #include <list>
  6. using namespace std;
  7. const int M=200010,N=(M<<1)+10;
  8. int n,x,y,pre[N],st[N],en[N],c[N];
  9. vector<int> u[N],v[N];
  10. long long ans[N];
  11. list<int> l[N];
  12. int dfn;
  13. void dfs(int x,int fx){
  14. int cx=c[x];
  15. if(l[cx].empty())u[M+cx].push_back(x);
  16. else u[l[cx].back()].push_back(x);
  17. pre[x]=fx;
  18. l[cx].push_back(x);
  19. st[x]=dfn++;
  20. for(int i=0;i<v[x].size();i++){
  21. int y=v[x][i];
  22. if(y==fx)continue;
  23. dfs(y,x);
  24. }l[cx].pop_back();
  25. en[x]=dfn;
  26. }
  27. bool isson(int x,int y){return st[y]<=st[x]&&st[x]<en[y];}
  28. void Dfs(int x,int d){
  29. int pos=0;
  30. if(x<=M){
  31. ans[st[x]]+=n-d;
  32. ans[st[x]+1]-=n-d;
  33. }
  34. for(int i=0;i<v[x].size();i++){
  35. int y=v[x][i];
  36. if(y==pre[x])continue;
  37. int p=pos,size=en[y]-st[y];
  38. while(p<u[x].size()&&isson(u[x][p],y)){
  39. size-=en[u[x][p]]-st[u[x][p]];
  40. p++;
  41. }ans[st[y]]+=n-size-d;
  42. ans[en[y]]-=n-size-d;
  43. for(int j=pos;j<p;j++)Dfs(u[x][j],n-size);
  44. pos=p;
  45. }
  46. }
  47. int Cas=1;
  48. int main(){
  49. while(~scanf("%d",&n)){
  50. for(int i=1;i<=n;i++)scanf("%d",&c[i]);
  51. memset(ans,0,sizeof(ans));
  52. for(int i=0;i<N;i++)v[i].clear(),u[i].clear(),l[i].clear();
  53. dfn=1;
  54. for(int i=1;i<n;i++){
  55. scanf("%d%d",&x,&y);
  56. v[x].push_back(y);
  57. v[y].push_back(x);
  58. }dfs(1,1);
  59. for(int i=1;i<=M;i++){v[i+M].push_back(1);Dfs(i+M,0);}
  60. for(int i=1;i<=n;i++)ans[i]+=ans[i-1];
  61. long long Ans=0;
  62. for(int i=1;i<=n;i++)Ans+=ans[st[i]];
  63. printf("Case #%d: %lld\n",Cas++,(Ans-n)/2);
  64. }return 0;
  65. }

HDU 6035 Colorful Tree (树形DP)的更多相关文章

  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. hdu 6035:Colorful Tree (2017 多校第一场 1003) 【树形dp】

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

  4. hdu6035 Colorful Tree 树形dp 给定一棵树,每个节点有一个颜色值。定义每条路径的值为经过的节点的不同颜色数。求所有路径的值和。

    /** 题目:hdu6035 Colorful Tree 链接:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题意:给定一棵树,每个节点有一个颜色值.定 ...

  5. 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 ...

  6. HDU-6035 Colorful Tree(树形DP) 2017多校第一场

    题意:给出一棵树,树上的每个节点都有一个颜色,定义一种值为两点之间路径中不同颜色的个数,然后一棵树有n*(n-1)/2条 路径,求所有的路径的值加起来是多少. 思路:比赛的时候感觉是树形DP,但是脑袋 ...

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

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

  8. HDU 6035 Colorful Tree(dfs)

    题意:一棵有n个点的树,树上每个点都有颜色c[i],定义每条路径的值为这条路径上经过的不同颜色数量和.求所有路径的值的和. 可以把问题转化为对每种颜色有多少条不同的路径至少经过这种颜色的点,然后加和. ...

  9. hdu 6035 Colorful Tree(虚树)

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

随机推荐

  1. SpringBoot Mybatis 读写分离配置(山东数漫江湖)

    为什么需要读写分离 当项目越来越大和并发越来大的情况下,单个数据库服务器的压力肯定也是越来越大,最终演变成数据库成为性能的瓶颈,而且当数据越来越多时,查询也更加耗费时间,当然数据库数据过大时,可以采用 ...

  2. 【洛谷 P4180】【模板】严格次小生成树[BJWC2010](倍增)

    题目链接 题意如题. 这题作为我们KS图论的T4,我直接打了个很暴力的暴力,骗了20分.. 当然,我们KS里的数据范围远不及这题. 这题我debug了整整一个晚上还没debug出来,第二天早上眼前一亮 ...

  3. 【洛谷 P5110】 块速递推(矩阵加速,分块打表)

    题目链接 掌握了分块打表法了.原来以前一直想错了... 块的大小\(size=\sqrt n\),每隔\(size\)个数打一个表,还要在\(0\text{~}size-1\)每个数打一个表. 然后就 ...

  4. ES6新特性学习(一)

    一.什么是ES6 ECMAScript和JavaScript的关系 ECMAScript 6(以下简称ES6)是JavaScript语言的下一代标准,已经在2015年6月正式发布了.Mozilla公司 ...

  5. js_md5加密和base64的加密解密

    1.最近有些人在爬我们公司的数据,然有了这个md5加密的小需求.为什么叫小需求呢?嗯,之前没做过,会以为很复杂. 其实,是想多了. 2.前端md5加密,其实也并不是安全的,因为代码是可见的.也就是说, ...

  6. javascript中=、==与===的区别

    1.等号 =赋值运算符,给变量赋值 var a="1"; 2.相等和不相等操作符 相等操作符由==表示,若两个操作数相等,则返回true:不相等操作符由!=表示,若两个操作数不相等 ...

  7. Java案例之士兵作战功能实现

    实现的功能比较简单,主要用到了多态的,抽象类以及模板方法模式这几个知识点.效果图如下,哈哈 ,大神勿喷,后面我会把这些知识点详细介绍出来,即使Java学的不好,只要有一点其他语言基础或者没有应该都能看 ...

  8. python模块中requests参数stream

    PS:这个参数真没用过 当下载大的文件的时候,建议使用strea模式. 默认情况下是false,他会立即开始下载文件并存放到内存当中,倘若文件过大就会导致内存不足的情况. 当把get函数的stream ...

  9. 【swupdate文档 一】嵌入式系统的软件管理

    嵌入式系统的软件管理 嵌入式系统变得越来越复杂, 它们的软件也反映了这种复杂性的增加. 为了支持新的特性和修复,很有必要让嵌入式系统上的软件 能够以绝对可靠的方式更新. 在基于linux的系统上,我们 ...

  10. .net爬虫了解一下

    using System; //添加selenium的引用 using OpenQA.Selenium.PhantomJS; using OpenQA.Selenium.Chrome; using O ...