hdu6035
树形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的更多相关文章
- hdu6035 Colorful Tree 树形dp 给定一棵树,每个节点有一个颜色值。定义每条路径的值为经过的节点的不同颜色数。求所有路径的值和。
/** 题目:hdu6035 Colorful Tree 链接:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题意:给定一棵树,每个节点有一个颜色值.定 ...
- hdu6035(树形DP)
hdu6035 题意 给出一棵树,现在定义两点之间距离为两点间最短路径上颜色集合的大小.问任意两点间距离之和. 分析 换个方向,题目其实等价于求每种颜色在多少条路径上出现过(每种颜色对于答案的贡献), ...
- hdu6035[dfs+思维] 2017多校1
/*hdu6035[dfs+思维] 2017多校1*/ //合并色块, 妙啊妙啊 #include<bits/stdc++.h> using namespace std; ; const ...
- HDU6035 Colorful Tree
题目链接:https://vjudge.net/problem/HDU-6035 题目大意: 多样例输入. 对于每一个样例,给出 n \((2 \le n \le 200000)\) 个结点的一棵树, ...
- hdu-6035 Colorful Tree
题目意思是计算所有路径(n*(n-1)/2)经过的不同颜色的数目和. 这个数目和可以转化为每种颜色经过的路径数目的求和,而这个求和又等价于颜色总数*n*(n-1)/2-没有经过某种颜色的边的数量的求和 ...
- 【hdu6035】 Colorful Tree dfs序
题目传送门:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题目大意:给你一棵树,树上每个节点都有一个颜色. 现在定义两点间的距离为两点最短路径上颜色集合 ...
- 2017 Multi-University Training Contest - Team 1—HDU6035
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题意:一棵树有n个点,每个点有自己的颜色,任意两个不同的点可以组成一条路径.也就是说一共有n(n ...
- 【树形dp】hdu6035 Colorful Tree
非常棒的题解,我就不复述了:http://blog.csdn.net/Bahuia/article/details/76141574 O(n) #include<cstdio> #incl ...
- HDU-6035:Colorful Tree(虚树+DP)
这里有三道长得像的题: 一:HDU6036: There is a tree with nn nodes, each of which has a type of color represented ...
- HDU-6035 Colorful Tree(树形DP) 2017多校第一场
题意:给出一棵树,树上的每个节点都有一个颜色,定义一种值为两点之间路径中不同颜色的个数,然后一棵树有n*(n-1)/2条 路径,求所有的路径的值加起来是多少. 思路:比赛的时候感觉是树形DP,但是脑袋 ...
随机推荐
- docker容器的导入导出
导出容器docker export 导出容器快照到本地文件$ sudo docker ps -aCONTAINER ID IMAGE COMMAND ...
- ubuntu 12.04 64bit 安装 teamviewer 8.0
1. 在http://www.teamviewer.com下载teamviewer_linux_x64.deb 2.sudo dpkg -i teamviewer_linux_x64.deb 3.如果 ...
- Swift--方法(函数)
方法是执行特殊任务的自包含代码块.你可以给方法名字来表示它的功能,而且在需要的时候调用这个名字的方法来执行它的任务. Swift方法的语法表达很灵活,从类似c的没有参数名的方法到oc复杂的带有名字和参 ...
- CodeForces 598C Nearest vectors
这题对精度要求很高.用atan2吧... #include<iostream> #include<cstring> #include<cmath> #include ...
- Topcoder SRM 657DIV2
前言: 像我这样一直在DIV2的弱菜..不知道说什么了. A:一定判断有8个‘R’,每行 每列只有一个 B题:大概是 int E,int EM,int M,int MH,int H 然后EM可以给值 ...
- 洛谷 P1731 [NOI1999]生日蛋糕
P1731 [NOI1999]生日蛋糕 题目背景 7月17日是Mr.W的生日,ACM-THU为此要制作一个体积为Nπ的M层 生日蛋糕,每层都是一个圆柱体. 设从下往上数第i(1<=i<=M ...
- 查看MySQL系统变量的命令
用了好长时间mysql,却没有用心记住一些有用的东西,加油! mysql> SHOW VARIABLES; +---------------------------------+-------- ...
- [Unit Testing] Mock an HTTP request using Nock while unit testing
When testing functions that make HTTP requests, it's not preferable for those requests to actually r ...
- 社交O2O的进化
引言 谁都想在O2O这个狂热的概念下分一杯羹,从O2O兴趣社交延伸到O2O生活服务,移动社交APP也是各显神通. 早在微信4.2版本号里,开机界面里那句"少发微信.多和朋友见见面" ...
- 【问题记录】LoadRunner 接口压测-json格式报文
[问题起因] 前段时间,协助其他项目录制接口压测脚本,对方要求请求报文内容实现参数化. 请求方法如下: 直接在Parameter List中新增一个parameter, 将请求报文放入dat文件中.这 ...