luogu P2664 树上游戏(点分治)
点分治真是一个好东西。可惜我不会
这种要求所有路经的题很可能是点分治。
然后我就不会了。。
既然要用点分治,就想,点分治有哪些优点?它可以\(O(nlogn)\)遍历分治树的所有子树。
那么现在的问题就是,如可快速(\(O(n)\)或O\((nlogn)\))求以一个点为根的时候,子树之间的贡献(当然还有根节点的)。
我们注意到一件事,就是一棵子树中一个点对其他子树的点产生贡献当且仅当这个点的颜色在它到根的路径上第一次出现(或者说只算上这些贡献答案正确),且贡献为以这个点为根的子树大小。(不考虑其它子树的颜色)
这个有什么用,我们可以遍历两遍子树,第一遍预处理出所有子树对其它子树的贡献(如上边一段所说把贡献统计),第二次遍历每一颗子树先把这颗树的贡献去掉,统计所有其它的树对这颗树的贡献。
那么具体该怎么做?
void calc(int u){
dfs1(u,0);
ans[u]+=sum;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(vis[v])continue;
cnt[a[u]]++;
sum-=size[v];color[a[u]]-=size[v];
change(v,u,-1);
cnt[a[u]]--;
tot=size[u]-size[v];
dfs2(v,u);
cnt[a[u]]++;
sum+=size[v];color[a[u]]+=size[v];
change(v,u,1);
cnt[a[u]]--;
}
clear(u,0);
}
首先dfs1是统计贡献的,用sum记录贡献和,color[i]记录第i种颜色的贡献。
然后根的答案就可以累加了。
那么如可判断一个颜色第一次出现?可以记录一个cnt[i]记录第i种颜色在到根的路径上出现多少次。当cnt[i]等于1的时候统计贡献。
然后
cnt[a[u]]++;
sum-=size[v];color[a[u]]-=size[v];
change(v,u,-1);
cnt[a[u]]--;
用来消除子树贡献。dfs2统计其它子树对这颗子树的贡献。
void dfs2(int u,int f){
cnt[a[u]]++;
if(cnt[a[u]]==1){
sum-=color[a[u]];
num++;
}
ans[u]+=sum+num*tot;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f||vis[v])continue;
dfs2(v,u);
}
if(cnt[a[u]]==1){
sum+=color[a[u]];
num--;
}
cnt[a[u]]--;
}
如果这颗子树中出现一个颜色,并且它是第一次出现,那么减去所有子树的color[a[u]],加上其它子树的节点总数,因为每一条到其它子树的路径都会产生贡献,这也是我们一开始不考虑贡献对其他子树影响的原因,因为遍历子树的时候会把这些重复的贡献减去。
更具体还是看代码。
// luogu-judger-enable-o2
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
#define int long long
const int N=101000;
int Cnt,head[N];
int g[N],size[N],cnt[N],a[N],sum,color[N],tot,num,root,all,vis[N],ans[N],n;
struct edge{
int to,nxt;
}e[N*2];
void add_edge(int u,int v){
Cnt++;
e[Cnt].nxt=head[u];
e[Cnt].to=v;
head[u]=Cnt;
}
int read(){
int sum=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){sum=sum*10+ch-'0';ch=getchar();}
return sum*f;
}
void getroot(int u,int f){
g[u]=0;size[u]=1;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f||vis[v])continue;
getroot(v,u);
size[u]+=size[v];
g[u]=max(g[u],size[v]);
}
g[u]=max(g[u],all-size[u]);
if(g[u]<g[root])root=u;
}
void dfs1(int u,int f){
cnt[a[u]]++;
size[u]=1;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f||vis[v])continue;
dfs1(v,u);
size[u]+=size[v];
}
if(cnt[a[u]]==1){
sum+=size[u];
color[a[u]]+=size[u];
}
cnt[a[u]]--;
}
void clear(int u,int f){
cnt[a[u]]++;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f||vis[v])continue;
clear(v,u);
}
if(cnt[a[u]]==1){
sum-=size[u];
color[a[u]]-=size[u];
}
cnt[a[u]]--;
}
void dfs2(int u,int f){
cnt[a[u]]++;
if(cnt[a[u]]==1){
sum-=color[a[u]];
num++;
}
ans[u]+=sum+num*tot;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f||vis[v])continue;
dfs2(v,u);
}
if(cnt[a[u]]==1){
sum+=color[a[u]];
num--;
}
cnt[a[u]]--;
}
void change(int u,int f,int k){
cnt[a[u]]++;
if(cnt[a[u]]==1){
sum+=k*size[u];color[a[u]]+=k*size[u];
}
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==f||vis[v])continue;
change(v,u,k);
}
cnt[a[u]]--;
}
void calc(int u){
dfs1(u,0);
ans[u]+=sum;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(vis[v])continue;
cnt[a[u]]++;
sum-=size[v];color[a[u]]-=size[v];
change(v,u,-1);
cnt[a[u]]--;
tot=size[u]-size[v];
dfs2(v,u);
cnt[a[u]]++;
sum+=size[v];color[a[u]]+=size[v];
change(v,u,1);
cnt[a[u]]--;
}
clear(u,0);
}
void work(int u){
calc(u);
vis[u]=1;
for(int i=head[u];i;i=e[i].nxt){
int v=e[i].to;
if(vis[v])continue;
root=0,all=size[v];
getroot(v,0);
work(root);
}
}
signed main(){
n=read();
for(int i=1;i<=n;i++)a[i]=read();
for(int i=1;i<n;i++){
int u=read(),v=read();
add_edge(u,v);add_edge(v,u);
}
g[0]=n+10;root=0;all=n;
getroot(1,0);work(root);
for(int i=1;i<=n;i++)printf("%lld\n",ans[i]);
return 0;
}
luogu P2664 树上游戏(点分治)的更多相关文章
- Luogu P2664 树上游戏 dfs+树上统计
题目: P2664 树上游戏 分析: 本来是练习点分治的时候看到了这道题.无意中发现题解中有一种方法可以O(N)解决这道题,就去膜拜了一下. 这个方法是,假如对于某一种颜色,将所有这种颜色的点全部删去 ...
- [LuoGu]P2664 树上游戏
Portal 这题真的好. 看到树上路径, 脑子里就要点分治 这一题对于每个点都要计算一遍, 如果暴算实在不好算, 这样我们就可以考虑算贡献. 直接计算每种颜色的贡献. 因为一条过重心的路径中, 可能 ...
- 洛谷P2664 树上游戏(点分治)
题意 题目链接 Sol 神仙题..Orz yyb 考虑点分治,那么每次我们只需要统计以当前点为\(LCA\)的点对之间的贡献以及\(LCA\)到所有点的贡献. 一个很神仙的思路是,对于任意两个点对的路 ...
- 洛谷P2664 树上游戏——点分治
原题链接 被点分治虐的心态爆炸了 题解 发现直接统计路径上的颜色数量很难,考虑转化一下统计方式.对于某一种颜色\(c\),它对一个点的贡献为从这个点出发且包含这种颜色的路径条数. 于是我们先点分一下, ...
- P2664 树上游戏
P2664 树上游戏 https://www.luogu.org/problemnew/show/P2664 分析: 点分治. 首先关于答案的统计转化成计算每个颜色的贡献. 1.计算从根出发的路径的答 ...
- 洛谷 P2664 树上游戏 解题报告
P2664 树上游戏 题目描述 \(\text{lrb}\)有一棵树,树的每个节点有个颜色.给一个长度为\(n\)的颜色序列,定义\(s(i,j)\) 为 \(i\) 到 \(j\) 的颜色数量.以及 ...
- 洛谷P2664 树上游戏(点分治)
传送门 题解 因为一个sb错误调了一个晚上……鬼晓得我为什么$solve(rt)$会写成$solve(v)$啊!!!一个$O(logn)$被我硬生生写成$O(n)$了竟然还能过$5$个点……话说还一直 ...
- 洛谷P2664 树上游戏 【点分治 + 差分】
题目 lrb有一棵树,树的每个节点有个颜色.给一个长度为n的颜色序列,定义s(i,j) 为i 到j 的颜色数量.以及 现在他想让你求出所有的sum[i] 输入格式 第一行为一个整数n,表示树节点的数量 ...
- 【洛谷P2664】 树上游戏 点分治
code: #include <bits/stdc++.h> #define N 200009 #define ll long long #define setIO(s) freopen( ...
随机推荐
- 面向对象和结构化程序设计的区别X
面向对象和结构化程序设计的区别 结构化程序的概念首先是从以往编程过程中无限制地使用转移语句而提出的.转移语句可以使程序的控制流程强制性的转向程序的任一处,在传统流程图中,就是用上节我们提到的" ...
- LayUI中实现上级下拉框动态加载下级下拉框js
js代码: var form = layui.form, layer = layui.layer; form.on("select(上级)", function(data){ va ...
- 粘包解决高端_Server
from socket import * #导入套接字模块的所有命令import subprocess #导入subprocess模块,用于执行命令行import struct #导入struck模块 ...
- BZOJ 1492 [NOI2007]货币兑换Cash (CDQ分治/splay 维护凸包)
题目大意:太长了略 splay调了两天一直WA弃疗了 首先,我们可以猜一个贪心,如果买/卖,就一定都买/卖掉,否则不买/卖 反正货币的行情都是已知的,没有任何风险,所以肯定要选择最最最优的方案了 容易 ...
- 代理上网环境配置docker私有库
最后更新时间:2018年12月27日 Docker使用代理上网去 pull 各类 images,需要做如下配置: 创建目录: /etc/systemd/system/docker.service.d ...
- poj 2663 Tri Tiling (状压dp+多米诺骨牌问题+滚动数组反思)
本来直接一波状压dpAC的 #include<cstdio> #include<cstring> #include<algorithm> #define REP(i ...
- poj 2288 Islands and Bridges (状压dp+Tsp问题)
这道题千辛万苦啊! 这道题要涉及到当前点和前面两个点,那就设dp[state][i][j]为当前状态为state,当前点为i,前一个点为j 这个状态表示和之前做炮兵那题很像,就是涉及到三个点时,就多设 ...
- OpenRail中地形模型特征的含义
点或点高程 点(附有 X.Y.Z 数据)与任何其他点之间没有功能关系.对开阔地形的随机测量拍摄可以被当做随机点的例子.点图元,如单元.圆圈和文字串是典型的 MICROSTATION 图元,用于用图表定 ...
- 2015 Multi-University Training Contest 4 hdu 5338 ZZX and Permutations
ZZX and Permutations Time Limit: 6000/3000 MS (Java/Others) Memory Limit: 131072/131072 K (Java/O ...
- 闲来无事爬了下通讯录 试手 jsdom
curl http://xxx.com/address/addresslist\?search\=%40 --cookie oa_cookie=123 -s| node parss .js js 代码 ...