题目链接

题意：统计树上每个结点中恰好出现了k次的颜色数。

dsu on tree/线段树合并裸题。

启发式合并1：(748ms)

 #include<bits/stdc++.h>

 using namespace std;

 typedef long long ll;

 const int N=1e5+;

 int n,m,k,a[N],b[N],nb,fa[N],son[N],siz[N],cnt[N],ans[N],now,ne,hd[N],ka;

 struct E {int v,nxt;} e[N<<];

 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}

 void dfs1(int u,int f) {

     fa[u]=f,son[u]=,siz[u]=;

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v==fa[u])continue;

         dfs1(v,u),siz[u]+=siz[v];

         if(siz[v]>siz[son[u]])son[u]=v;

     }

 }

 void add(int x,int dx) {

     if(cnt[x]==k)--now;

     cnt[x]+=dx;

     if(cnt[x]==k)++now;

 }

 void cal(int u,int x) {

     add(a[u],x);

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v!=fa[u])cal(v,x);

     }

 }

 void dfs2(int u,int f) {

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v!=fa[u]&&v!=son[u])dfs2(v,);

     }

     if(son[u])dfs2(son[u],);

     add(a[u],);

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v!=fa[u]&&v!=son[u])cal(v,);

     }

     ans[u]=now;

     if(!f) {

         add(a[u],-);

         for(int i=hd[u]; ~i; i=e[i].nxt) {

             int v=e[i].v;

             if(v!=fa[u])cal(v,-);

         }

     }

 }

 int main() {

     int T;

     for(scanf("%d",&T); T--;) {

         if(ka)puts("");

         printf("Case #%d:\n",++ka);

         memset(hd,-,sizeof hd),ne=;

         memset(cnt,,sizeof cnt);

         memset(ans,,sizeof ans),now=;

         scanf("%d%d",&n,&k);

         for(int i=; i<=n; ++i)scanf("%d",&a[i]);

         for(int i=; i<=n; ++i)b[i]=a[i];

         sort(b+,b++n),nb=unique(b+,b++n)-(b+);

         for(int i=; i<=n; ++i)a[i]=lower_bound(b+,b++nb,a[i])-b;

         for(int i=; i<n; ++i) {

             int u,v;

             scanf("%d%d",&u,&v);

             addedge(u,v),addedge(v,u);

         }

         dfs1(,-),dfs2(,);

         scanf("%d",&m);

         while(m--) {

             int u;

             scanf("%d",&u);

             printf("%d\n",ans[u]);

         }

     }

     return ;

 }

启发式合并2(加了dfs序的启发式合并，只比普通的快了一丁点)：(702ms)

 #include<bits/stdc++.h>

 using namespace std;

 typedef long long ll;

 const int N=1e5+;

 int n,m,k,a[N],b[N],nb,fa[N],son[N],siz[N],cnt[N],ans[N],now,ne,hd[N],ka,tot,bg[N],ed[N],rnk[N];

 struct E {int v,nxt;} e[N<<];

 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}

 void dfs1(int u,int f) {

     fa[u]=f,son[u]=,siz[u]=,bg[u]=++tot,rnk[bg[u]]=u;

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v==fa[u])continue;

         dfs1(v,u),siz[u]+=siz[v];

         if(siz[v]>siz[son[u]])son[u]=v;

     }

     ed[u]=tot;

 }

 void add(int x,int dx) {

     if(cnt[x]==k)--now;

     cnt[x]+=dx;

     if(cnt[x]==k)++now;

 }

 void dfs2(int u,int f) {

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v!=fa[u]&&v!=son[u])dfs2(v,);

     }

     if(son[u])dfs2(son[u],);

     add(a[u],);

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v!=fa[u]&&v!=son[u])for(int i=bg[v]; i<=ed[v]; ++i)add(a[rnk[i]],);

     }

     ans[u]=now;

     if(!f)for(int i=bg[u]; i<=ed[u]; ++i)add(a[rnk[i]],-);

 }

 int main() {

     int T;

     for(scanf("%d",&T); T--;) {

         if(ka)puts("");

         printf("Case #%d:\n",++ka);

         memset(hd,-,sizeof hd),ne=;

         memset(cnt,,sizeof cnt),tot=;

         memset(ans,,sizeof ans),now=;

         scanf("%d%d",&n,&k);

         for(int i=; i<=n; ++i)scanf("%d",&a[i]);

         for(int i=; i<=n; ++i)b[i]=a[i];

         sort(b+,b++n),nb=unique(b+,b++n)-(b+);

         for(int i=; i<=n; ++i)a[i]=lower_bound(b+,b++nb,a[i])-b;

         for(int i=; i<n; ++i) {

             int u,v;

             scanf("%d%d",&u,&v);

             addedge(u,v),addedge(v,u);

         }

         dfs1(,-),dfs2(,);

         scanf("%d",&m);

         while(m--) {

             int u;

             scanf("%d",&u);

             printf("%d\n",ans[u]);

         }

     }

     return ;

 }

启发式合并3(map版，原理与普通启发式合并相同，轻链合并到重链上，重链中的父结点的贡献直接加进子结点)：(1185ms)

 #include<bits/stdc++.h>

 using namespace std;

 typedef long long ll;

 const int N=1e5+;

 int n,m,k,a[N],b[N],F[N],nb,ans[N],ne,hd[N],ka,fa[N],son[N],siz[N];

 map<int,int> mp[N];

 struct E {int v,nxt;} e[N<<];

 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}

 void add(int u,int x,int dx) {

     if(mp[F[u]][x]==k)--ans[u];

     mp[F[u]][x]+=dx;

     if(mp[F[u]][x]==k)++ans[u];

 }

 void mg(int u,int v) {

     for(auto p:mp[F[v]])add(u,p.first,p.second);

     mp[F[v]].clear();

 }

 void dfs1(int u,int f) {

     fa[u]=f,son[u]=,siz[u]=;

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v==fa[u])continue;

         dfs1(v,u),siz[u]+=siz[v];

         if(siz[v]>siz[son[u]])son[u]=v;

     }

 }

 void dfs2(int u) {

     F[u]=u;

     if(son[u])dfs2(son[u]),ans[u]=ans[son[u]],F[u]=F[son[u]];

     add(u,a[u],);

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v==fa[u]||v==son[u])continue;

         dfs2(v),mg(u,v);

     }

 }

 int main() {

     int T;

     for(scanf("%d",&T); T--;) {

         if(ka)puts("");

         printf("Case #%d:\n",++ka);

         memset(hd,-,sizeof hd),ne=;

         memset(ans,,sizeof ans);

         scanf("%d%d",&n,&k);

         for(int i=; i<=n; ++i)scanf("%d",&a[i]);

         for(int i=; i<=n; ++i)b[i]=a[i];

         sort(b+,b++n),nb=unique(b+,b++n)-(b+);

         for(int i=; i<=n; ++i)a[i]=lower_bound(b+,b++nb,a[i])-b;

         for(int i=; i<n; ++i) {

             int u,v;

             scanf("%d%d",&u,&v);

             addedge(u,v),addedge(v,u);

         }

         dfs1(,-),dfs2(),mp[F[]].clear();

         scanf("%d",&m);

         while(m--) {

             int u;

             scanf("%d",&u);

             printf("%d\n",ans[u]);

         }

     }

     return ;

 }

启发式合并4(map版，直接根据子树大小进行合并)：(1263ms)

 #include<bits/stdc++.h>

 using namespace std;

 typedef long long ll;

 const int N=1e5+;

 int n,m,k,a[N],b[N],F[N],nb,ans[N],ne,hd[N],ka;

 map<int,int> mp[N];

 struct E {int v,nxt;} e[N<<];

 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}

 void add(int u,int x,int dx) {

     if(mp[F[u]][x]==k)--ans[u];

     mp[F[u]][x]+=dx;

     if(mp[F[u]][x]==k)++ans[u];

 }

 void mg(int u,int v) {

     if(mp[F[u]].size()<mp[F[v]].size())ans[u]=ans[v],swap(F[u],F[v]);

     for(auto p:mp[F[v]])add(u,p.first,p.second);

     mp[F[v]].clear();

 }

 void dfs(int u,int fa) {

     add(F[u]=u,a[u],);

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v==fa)continue;

         dfs(v,u),mg(u,v);

     }

 }

 int main() {

     int T;

     for(scanf("%d",&T); T--;) {

         if(ka)puts("");

         printf("Case #%d:\n",++ka);

         memset(hd,-,sizeof hd),ne=;

         memset(ans,,sizeof ans);

         scanf("%d%d",&n,&k);

         for(int i=; i<=n; ++i)scanf("%d",&a[i]);

         for(int i=; i<=n; ++i)b[i]=a[i];

         sort(b+,b++n),nb=unique(b+,b++n)-(b+);

         for(int i=; i<=n; ++i)a[i]=lower_bound(b+,b++nb,a[i])-b;

         for(int i=; i<n; ++i) {

             int u,v;

             scanf("%d%d",&u,&v);

             addedge(u,v),addedge(v,u);

         }

         dfs(,-),mp[F[]].clear();

         scanf("%d",&m);

         while(m--) {

             int u;

             scanf("%d",&u);

             printf("%d\n",ans[u]);

         }

     }

     return ;

 }

线段树合并：(1014ms)

 #include<bits/stdc++.h>

 using namespace std;

 typedef long long ll;

 const int N=1e5+;

 int n,m,k,a[N],b[N],nb,ans[N],ne,hd[N],ka,tot,rt[N],ls[N*],rs[N*],val[N*],sum[N*];

 #define mid ((l+r)>>1)

 struct E {int v,nxt;} e[N<<];

 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}

 void pu(int u) {sum[u]=sum[ls[u]]+sum[rs[u]];}

 int newnode() {int u=++tot; ls[u]=rs[u]=val[u]=sum[u]=; return u;}

 void add(int& u,int x,int dx,int l=,int r=nb) {

     if(!u)u=newnode();

     if(l==r) {val[u]+=dx,sum[u]=val[u]==k; return;}

     x<=mid?add(ls[u],x,dx,l,mid):add(rs[u],x,dx,mid+,r);

     pu(u);

 }

 void mg(int& u,int v,int l=,int r=nb) {

     if(!u||!v) {u=u|v; return;}

     if(l==r) {val[u]+=val[v],sum[u]=val[u]==k; return;}

     mg(ls[u],ls[v],l,mid),mg(rs[u],rs[v],mid+,r),pu(u);

 }

 void dfs(int u,int fa) {

     rt[u]=,add(rt[u],a[u],);

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v==fa)continue;

         dfs(v,u),mg(rt[u],rt[v]);

     }

     ans[u]=sum[rt[u]];

 }

 int main() {

     int T;

     for(scanf("%d",&T); T--;) {

         if(ka)puts("");

         printf("Case #%d:\n",++ka);

         memset(hd,-,sizeof hd),ne=;

         memset(ans,,sizeof ans),tot=;

         scanf("%d%d",&n,&k);

         for(int i=; i<=n; ++i)scanf("%d",&a[i]);

         for(int i=; i<=n; ++i)b[i]=a[i];

         sort(b+,b++n),nb=unique(b+,b++n)-(b+);

         for(int i=; i<=n; ++i)a[i]=lower_bound(b+,b++nb,a[i])-b;

         for(int i=; i<n; ++i) {

             int u,v;

             scanf("%d%d",&u,&v);

             addedge(u,v),addedge(v,u);

         }

         dfs(,-);

         scanf("%d",&m);

         while(m--) {

             int u;

             scanf("%d",&u);

             printf("%d\n",ans[u]);

         }

     }

     return ;

 }

还不够爽？再来个资瓷在线查询的可持久化线段树合并怎么样？就是有点吃内存。(1310ms)

 #include<bits/stdc++.h>

 using namespace std;

 typedef long long ll;

 const int N=1e5+;

 int n,m,k,a[N],b[N],nb,ne,hd[N],ka,tot,rt[N],ls[N*],rs[N*],val[N*],sum[N*];

 #define mid ((l+r)>>1)

 struct E {int v,nxt;} e[N<<];

 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}

 void pu(int u) {sum[u]=sum[ls[u]]+sum[rs[u]];}

 int newnode() {int u=++tot; ls[u]=rs[u]=val[u]=sum[u]=; return u;}

 void add(int& w,int u,int x,int dx,int l=,int r=nb) {

     w=newnode();

     if(l==r) {val[w]=val[u]+dx,sum[w]=val[w]==k; return;}

     if(x<=mid)add(ls[w],ls[u],x,dx,l,mid),rs[w]=rs[u];

     else add(rs[w],rs[u],x,dx,mid+,r),ls[w]=ls[u];

     pu(w);

 }

 void mg(int& w,int u,int v,int l=,int r=nb) {

     if(!u||!v) {w=u|v; return;}

     w=newnode();

     if(l==r) {val[w]=val[u]+val[v],sum[w]=val[w]==k; return;}

     mg(ls[w],ls[u],ls[v],l,mid),mg(rs[w],rs[u],rs[v],mid+,r),pu(w);

 }

 void dfs(int u,int fa) {

     rt[u]=,add(rt[u],rt[u],a[u],);

     for(int i=hd[u]; ~i; i=e[i].nxt) {

         int v=e[i].v;

         if(v==fa)continue;

         dfs(v,u),mg(rt[u],rt[u],rt[v]);

     }

 }

 int main() {

     int T;

     for(scanf("%d",&T); T--;) {

         if(ka)puts("");

         printf("Case #%d:\n",++ka);

         memset(hd,-,sizeof hd),ne=;

         tot=;

         scanf("%d%d",&n,&k);

         for(int i=; i<=n; ++i)scanf("%d",&a[i]);

         for(int i=; i<=n; ++i)b[i]=a[i];

         sort(b+,b++n),nb=unique(b+,b++n)-(b+);

         for(int i=; i<=n; ++i)a[i]=lower_bound(b+,b++nb,a[i])-b;

         for(int i=; i<n; ++i) {

             int u,v;

             scanf("%d%d",&u,&v);

             addedge(u,v),addedge(v,u);

         }

         dfs(,-);

         scanf("%d",&m);

         while(m--) {

             int u;

             scanf("%d",&u);

             printf("%d\n",sum[rt[u]]);

         }

     }

     return ;

 }

HDU - 4358 Boring counting (树上启发式合并/线段树合并)的更多相关文章

  1. 洛谷P3605 [USACO17JAN] Promotion Counting 晋升者计数 [线段树合并]

    题目传送门 Promotion Counting 题目描述 The cows have once again tried to form a startup company, failing to r ...

  2. hdu 5511 Minimum Cut-Cut——分类讨论思想+线段树合并

    题目:http://acm.hdu.edu.cn/showproblem.php?pid=5511 题意:割一些边使得无向图变成不连通的,并且恰好割了两条给定生成树上的边.满足非树边两段一定在给定生成 ...

  3. 启发式合并 splay合并 线段树合并基础

    Gold is everywhen! - somebody 启发式合并 将小的集合一个个插入到大的集合. 每次新集合大小至少比小集合大一倍,因此每个元素最多合并\(\log n\)次,总复杂度为\(n ...

  4. 启发式合并&线段树合并/分裂&treap合并&splay合并

    启发式合并 有\(n\)个集合,每次让你合并两个集合,或询问一个集合中是否存在某个元素. ​ 我们可以用平衡树/set维护集合. ​ 对于合并两个\(A,B\),如果\(|A|<|B|\),那么 ...

  5. [BZOJ2733][HNOI2010]永无乡 解题报告 启发式合并,线段树合并

    好久没更新博客了,前段时间一直都在考试,都没时间些,现在终于有点闲了(cai guai)... 写了一道题,[HNOI2012]永无乡,其实是一道板子题,我发现我写了好多板子题...还是太菜了... ...

  6. bzoj2733: [HNOI2012]永无乡（splay+启发式合并/线段树合并）

    这题之前写过线段树合并,今天复习Splay的时候想起这题,打算写一次Splay+启发式合并. 好爽!!! 写了长长的代码(其实也不长),只凭着下午的一点记忆(没背板子...),调了好久好久,过了样例, ...

  7. Luogu5327【ZJOI2019】语言【树上差分，线段树合并】

    题目大意 给定一棵$n$个节点的树,维护$n$个集合,一开始第$i$个集合只有节点$i$.有$m$个操作,每次操作输入一个$(u,v)$,表示将$(u,v)$这条链上所有点所属的集合合并.求有多少个无 ...

  8. 5.20 省选模拟赛 T1 图 启发式合并 线段树合并 染色计数问题

    LINK:图 在说这道题之前吐槽一下今天的日子 520 = 1+1+4+514. /cy 这道题今天做的非常失败 一点分都没拿到手 关键是今天的T3 把我整个人给搞崩了. 先考虑 如果得到了这么一张图 ...

  9. HDU 4358 Boring counting（莫队+DFS序+离散化）

    Boring counting Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 98304/98304 K (Java/Others) ...

随机推荐

  1. CompletableFuture引入

    一.Future介绍 Future以前我们如果有个方法action执行,比如去数据库中查询数据.上传一个文件等,这个方法执行10分钟,调用者就需要等10分钟.基于此,调用者可以先执行action,返回 ...

  2. rocketMQ broker 分发并处理请求

    使用 netty 监听端口 // org.apache.rocketmq.remoting.netty.NettyRemotingServer#start ServerBootstrap childH ...

  3. 考虑以下 Python 代码，如果运行结束，命令行中的运行结果是什么？

    l = [] for i in xrange(10):  l.append({‘num’:i}) print l在考虑以下代码,运行结束后的结果是什么? l = [] a = {‘num’:0} fo ...

  4. asp.net mvc中用 log4net记录日志到数据库中

    1.log4net官网配置相关,创建数据库 http://logging.apache.org/log4net/release/config-examples.html CREATE TABLE [d ...

  5. mycat是什么？你是怎么理解的？你们公司分库分表的分片规则是什么？搭建mycat环境常用的配置文件有哪些？

    1.mycat是什么? 国内最活跃的.性能最好的开源数据库分库分表中间件 一个彻底开源的,面向企业应用开发的大数据库集群 支持事务.ACID.可以替代MySQL的加强版数据库 一个可以视为MySQL集 ...

  6. 【Qt开发】在QLabel已经显示背景图片后绘制图形注意事项

    主要是要解决图形覆盖的问题,通常的办法就是对QLabel进行子类化,并重载函数: void myLabel::paintEvent(QPaintEvent *event)   {       QLab ...

  7. Revo Uninstaller Pro - 真正彻底卸载软件不留垃圾的强大神器！(清理安装残留文件/注册表)

    大家都知道 Windows 在卸载软件时总是不够彻底,系统C盘总会留下大量难以辨别和清理的垃圾文件和临时文件,时间长了注册表也会变得非常臃肿,不仅浪费硬盘空间,而且也会明显拖慢系统响应和启动速度. R ...

  8. Java第六周实验+总结

    一.实验目的 (1)掌握类的继承 1.子类继承父类中非private的成员变量和成员方法,同时,注意构造方法不能被子类继承. 2.定义类时若缺省extends关键字,则所定义的类为java.lang. ...

  9. WebMagic 抓取图片并保存至本地

    1.近期接触到java 爬虫,开源的爬虫框架有很多,其中WebMagic 是国产的,文档也是中文的,网上资料很多,便于学习,功能强大,可以在很短时间内实现一个简单的网络爬虫.具体可参考官网 http: ...

  10. [转帖]oracle备份恢复之recover database的四条语句区别

    oracle备份恢复之recover database的四条语句区别 https://www.cnblogs.com/andy6/p/5925433.html 需要学习一下. 1  recover d ...