[SPOJ]COT2
题意:给一棵带点权的树,多次询问两点间路径上的不同权值数
学习了一下莫队上树(雾
先求出栈入栈序$p_{1\cdots 2n}$,记$st_x$为$x$在$p$中第一次出现的位置,$ed_x$为$x$在$p$中最后一次出现的位置
对于一个询问$(x,y)$,先令$st_x\lt st_y$,求出其$lca$,若$x=lca$,则询问$[st_x,st_y]$,否则询问$[ed_x,st_y]$还有$lca$(因为$[ed_x,st_y]$不包含$lca$)
于是我们成功地把问题转化到序列上,普通地莫队即可
不应处理出现两次的数(因为它入栈一次,出栈一次,在路径之外)
#include<stdio.h> #include<algorithm> #include<map> #include<math.h> using namespace std; struct ask{ int l,r,lca,sid,qid; }q[100010]; int to[80010],nex[80010],h[40010],dep[40010],fa[40010][16],v[40010],p[80010],st[40010],ed[40010],ti[40010],ex[40010],ans[100010],M; map<int,int>mp; map<int,int>::iterator it; bool cmp(ask a,ask b){ if(a.sid==b.sid)return a.r<b.r; return a.sid<b.sid; } void add(int a,int b){ M++; to[M]=b; nex[M]=h[a]; h[a]=M; } void dfs(int x){ st[x]=++M; p[M]=x; for(int i=h[x];i;i=nex[i]){ if(to[i]!=fa[x][0]){ fa[to[i]][0]=x; dep[to[i]]=dep[x]+1; dfs(to[i]); } } ed[x]=++M; p[M]=x; } int lca(int x,int y){ if(dep[x]<dep[y])swap(x,y); int i; for(i=15;i>=0;i--){ if(dep[fa[x][i]]>=dep[y])x=fa[x][i]; } if(x==y)return x; for(i=15;i>=0;i--){ if(fa[x][i]!=fa[y][i]){ x=fa[x][i]; y=fa[y][i]; } } return fa[x][0]; } int sum; void add(int); void del(int x){ if(ex[x]==0)return add(x); ti[v[x]]--; if(ti[v[x]]==0)sum--; ex[x]=0; } void add(int x){ if(ex[x])return del(x); if(ti[v[x]]==0)sum++; ti[v[x]]++; ex[x]=1; } int main(){ int n,m,i,j,x,y,l,r,sq; scanf("%d%d",&n,&m); sq=sqrt(n); for(i=1;i<=n;i++){ scanf("%d",v+i); mp[v[i]]=1; } for(i=1,it=mp.begin();it!=mp.end();it++,i++)it->second=i; for(i=1;i<=n;i++)v[i]=mp[v[i]]; for(i=1;i<n;i++){ scanf("%d%d",&x,&y); add(x,y); add(y,x); } M=0; dep[1]=1; dfs(1); for(j=1;j<16;j++){ for(i=1;i<=n;i++)fa[i][j]=fa[fa[i][j-1]][j-1]; } for(i=1;i<=m;i++){ scanf("%d%d",&x,&y); if(st[x]>st[y])swap(x,y); j=lca(x,y); q[i].r=st[y]; if(x==j) q[i].l=st[x]; else{ q[i].l=ed[x]; q[i].lca=j; } q[i].sid=q[i].l/sq; q[i].qid=i; } sort(q+1,q+m+1,cmp); l=r=1; add(1); for(i=1;i<=m;i++){ while(l>q[i].l){ l--; add(p[l]); } while(r<q[i].r){ r++; add(p[r]); } while(l<q[i].l){ del(p[l]); l++; } while(r>q[i].r){ del(p[r]); r--; } if(q[i].lca)add(q[i].lca); ans[q[i].qid]=sum; if(q[i].lca)del(q[i].lca); } for(i=1;i<=m;i++)printf("%d\n",ans[i]); }
[SPOJ]COT2的更多相关文章
- SPOJ COT2 - Count on a tree II(LCA+离散化+树上莫队)
COT2 - Count on a tree II #tree You are given a tree with N nodes. The tree nodes are numbered from ...
- SPOJ COT2 Count on a tree II(树上莫队)
题目链接:http://www.spoj.com/problems/COT2/ You are given a tree with N nodes.The tree nodes are numbere ...
- spoj COT2 - Count on a tree II
COT2 - Count on a tree II http://www.spoj.com/problems/COT2/ #tree You are given a tree with N nodes ...
- SPOJ COT2 Count on a tree II (树上莫队)
题目链接:http://www.spoj.com/problems/COT2/ 参考博客:http://www.cnblogs.com/xcw0754/p/4763804.html上面这个人推导部分写 ...
- SPOJ COT2 树上找路径上不同值的个数
题目大意 给出多个询问u , v , 求出u-v路径上点权值不同的个数 开始做的是COT1,用主席树写过了,理解起来不难 很高兴的跑去做第二道,完全跟普通数组区间求k个不同有很大区别,完全没思路 膜拜 ...
- spoj COT2 - Count on a tree II 树上莫队
题目链接 http://codeforces.com/blog/entry/43230树上莫队从这里学的, 受益匪浅.. #include <iostream> #include < ...
- spoj COT2(树上莫队)
模板.树上莫队的分块就是按dfn分,然后区间之间转移时注意一下就好.有个图方便理解http://blog.csdn.net/thy_asdf/article/details/47377709: #in ...
- 「日常训练&知识学习」莫队算法(二):树上莫队(Count on a tree II,SPOJ COT2)
题意与分析 题意是这样的,给定一颗节点有权值的树,然后给若干个询问,每次询问让你找出一条链上有多少个不同权值. 写这题之前要参看我的三个blog:Codeforces Round #326 Div. ...
- SPOJ - COT2 离线路径统计
题意:求\(u\)到\(v\)的最短路径的不同权值种类个数 树上莫队试水题,这一篇是上篇的弱化部分,但可测试以下结论的正确性 设\(S(u,v)\):\(u-v\)最短路径所覆盖的点集 \(S(u,v ...
随机推荐
- ng的ngModel用来处理表单操作
https://segmentfault.com/a/1190000009126012
- WITH AS 使用
WITH AS 含义: WITH AS短语,也叫做子查询部分(subquery factoring),可以让你做很多事情,定义一个SQL片断,该SQL片断会被整个SQL语句所用到.有的时候,是为了让S ...
- Spring学习--集合属性
Spring 中可以通过一组内置的 xml 标签(例如: <list> , <set> 或 <map>) 来配置集合属性. 配置java.util.Set 需要使用 ...
- bootstrap再次回顾认识到的东西
1,需要使用html5文档类型(Doctype),因此在使用bootstrap项目的开头包含下面的代码段. <!DOCTYPE html> <html> ....... < ...
- mybatis 关系映射
一:订单商品数据模型 1.数据库执行脚本 创建数据库表代码: 1 CREATE TABLE items ( 2 id INT NOT NULL AUTO_INCREMENT, 3 itemsname ...
- lesson 4 再谈继承多态,抽象类和接口
再谈多态,抽象类和接口 上一次博客已经概念性的概述了继承多态,抽象类和接口,这次来具体的谈一谈他们之间的联系和需要注意的地方. 一.继承和多态:Inheritance (继承) & Polym ...
- sql 批量更新表中多字段为不同的值
,),,),rand()) select newid() ,) update tablename , FB,)) , ), FC,)) , )
- xcode自动打ipa包脚本
http://blog.csdn.net/ccf0703/article/details/7999112 文章首发地址:http://webfrogs.github.com/IOS/2012/09/1 ...
- KVM分析报告
转载 KVM分析报告 虚拟化技术工作组 2008-12-31 1. 概述 1.1. KVM简介 KVM是以色列开源组织Qumranet开发的一个开源虚拟机监控器,从Linux-2. ...
- container_of分析--可用good【转】
转自:http://blog.csdn.net/tigerjibo/article/details/8299589 版权声明:本文为博主原创文章,未经博主允许不得转载. 1.container_of宏 ...