树形dp

并没有想出来。。。

这个点分治很明显做不了,没办法批量处理信息,那么我们就只能考虑树形dp了,然后觉得用虚树什么的每个颜色单独考虑,GG

事实上我们可以这样考虑,我们对树进行dfs,每个节点有一个颜色,那么我们考虑子树中有多少条路径不包含这个颜色,很明显这条路径不能路过这个点,也就是说答案是C(n,2),n是每个儿子中不包含这个颜色的连通块大小,具体看程序,这里我们利用线段树合并来计算连通块大小,也可以用差分(?)统计一下就行了,在1号节点时要把除了1号节点的颜色也算上,然后用总路径条数减去即可

大佬还是强啊。。。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = ;
vector<int> G[N];
int n, cnt, color;
ll ans;
int size[N], sum[N * ], lc[N * ], rc[N * ], root[N], c[N], vis[N];
namespace Segment_Tree
{
void update(int l, int r, int &o, int pos, int d)
{
o = ++cnt;
sum[o] += d;
if(l == r) return;
int mid = (l + r) >> ;
if(pos <= mid) update(l, mid, lc[o], pos, d);
else update(mid + , r, rc[o], pos, d);
}
int merge(int u, int v)
{
if(!u) return v;
if(!v) return u;
lc[u] = merge(lc[u], lc[v]);
rc[u] = merge(rc[u], rc[v]);
sum[u] += sum[v];
return u;
}
int query(int l, int r, int o, int pos)
{
if(!o) return ;
if(l == r) return sum[o];
int mid = (l + r) >> ;
if(pos <= mid) return query(l, mid, lc[o], pos);
else return query(mid + , r, rc[o], pos);
}
} using namespace Segment_Tree;
void dfs(int u, int last)
{
size[u] = ;
ll tot = ;
for(int i = ; i < G[u].size(); ++i)
{
int v = G[u][i];
if(v == last) continue;
dfs(v, u);
size[u] += size[v];
ll tmp = query(, n, root[v], c[u]);
tot += tmp;
ans -= (size[v] - tmp) * (size[v] - tmp - ) / 2ll;
}
update(, n, root[u], c[u], size[u] - tot);
for(int i = ; i < G[u].size(); ++i)
{
int v = G[u][i];
if(v == last) continue;
root[u] = merge(root[u], root[v]);
}
}
int main()
{
scanf("%d", &n);
ans = (ll)n * (ll)(n - ) / 2ll;
for(int i = ; i <= n; ++i) scanf("%d", &c[i]), vis[c[i]] = ;
for(int i = ; i <= n; ++i) color += vis[i];
ans *= color;
for(int i = ; i < n; ++i)
{
int u, v;
scanf("%d%d", &u, &v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(, );
for(int i = ; i <= n; ++i) if(vis[i] && c[] != i)
{
ll tmp = query(, n, root[], i);
ans -= (ll)(size[] - tmp) * (ll)(size[] - tmp - ) / 2ll;
}
printf("%lld\n", ans);
return ;
}

hdu6035的更多相关文章

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

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

  2. hdu6035(树形DP)

    hdu6035 题意 给出一棵树,现在定义两点之间距离为两点间最短路径上颜色集合的大小.问任意两点间距离之和. 分析 换个方向,题目其实等价于求每种颜色在多少条路径上出现过(每种颜色对于答案的贡献), ...

  3. hdu6035[dfs+思维] 2017多校1

    /*hdu6035[dfs+思维] 2017多校1*/ //合并色块, 妙啊妙啊 #include<bits/stdc++.h> using namespace std; ; const ...

  4. HDU6035 Colorful Tree

    题目链接:https://vjudge.net/problem/HDU-6035 题目大意: 多样例输入. 对于每一个样例,给出 n \((2 \le n \le 200000)\) 个结点的一棵树, ...

  5. hdu-6035 Colorful Tree

    题目意思是计算所有路径(n*(n-1)/2)经过的不同颜色的数目和. 这个数目和可以转化为每种颜色经过的路径数目的求和,而这个求和又等价于颜色总数*n*(n-1)/2-没有经过某种颜色的边的数量的求和 ...

  6. 【hdu6035】 Colorful Tree dfs序

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

  7. 2017 Multi-University Training Contest - Team 1—HDU6035

    题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题意:一棵树有n个点,每个点有自己的颜色,任意两个不同的点可以组成一条路径.也就是说一共有n(n ...

  8. 【树形dp】hdu6035 Colorful Tree

    非常棒的题解,我就不复述了:http://blog.csdn.net/Bahuia/article/details/76141574 O(n) #include<cstdio> #incl ...

  9. HDU-6035:Colorful Tree(虚树+DP)

    这里有三道长得像的题: 一:HDU6036: There is a tree with nn nodes, each of which has a type of color represented ...

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

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

随机推荐

  1. Monkey King(左偏树)

    洛谷传送门 每次给出要争吵的猴子a和b,用并查集判断如果他们是朋友输出-1 如果不是,找出a,b在的堆的根A,B,分别合并A,B的左右孩子,再合并一下. 之后把A,B的数据更改一下:权值除以2,左右孩 ...

  2. hud 2073

    #include<stdio.h> #include<math.h> int main() { int i,j,k,n,m,t; double a]; a; for;i++) ...

  3. bzoj3514 Codechef MARCH14 GERALD07加强版 lct预处理+主席树

    Codechef MARCH14 GERALD07加强版 Time Limit: 60 Sec  Memory Limit: 256 MBSubmit: 1951  Solved: 746[Submi ...

  4. Python的另一种开发环境--Anaconda中的Spyder

    本文作者LucyGill,转载请注明出处(虽然我觉得并不会有人转载). 刚开始学Python的时候,我用的是其自带的idle(安装Python后,在开始菜单里可以找到),后来发现在eclipse中设置 ...

  5. HDU 6395 分段矩阵快速幂 HDU 6386 建虚点+dij

    http://acm.hdu.edu.cn/showproblem.php?pid=6395 Sequence Time Limit: 4000/2000 MS (Java/Others)    Me ...

  6. 1370 - Bi-shoe and Phi-shoe(LightOJ1370)(数论基础,欧拉函数)

    http://lightoj.com/volume_showproblem.php?problem=1370 欧拉函数: 在数论,对正整数n,欧拉函数是少于或等于n的数中与n互质的数的数目. φ(n) ...

  7. Javascript小数取整方法收集

    1.丢弃小数部分,保留整数部分 parseInt(7/2) 2.向上取整,有小数就整数部分加1 Math.ceil(7/2) 3.四舍五入 Math.round(7/2) 4.向下取整 Math.fl ...

  8. maven打包时的三方包的选择顺序

    在一个项目有多个模块引用多个版本的某个插件(或者叫三方包.jar包等)时,如何解决版本冲突问题?最终选用某个版本还是选择几个版本? maven在遇到上面的情况时,会智能处理版本冲突,最终选择一个版本, ...

  9. 【转载】《Unix网络编程》思维导图

    参考这篇文章,很不错: http://www.cnblogs.com/qiaoconglovelife/p/5734768.html

  10. eclipse环境下无法创建android virtual Devices(AVD)问题解决的方法汇总

    首先,要在eclipse环境下成功的创建一个安卓虚拟机,须要有三项东西,第一就是eclipse,第二就是android SDK Manager,第三就是ADT,也就是eclipse环境下的一个安卓虚拟 ...