题目链接

BZOJ4539

题解

我们把每次复制出来的树看做一个点,那么大树实际上也就是一棵\(O(M)\)个点的树

所以我们只需求两遍树上距离:

大树上求距离,进入同一个点后在模板树上再求一次距离

讨论好一些情况即可

然后求子树第\(k\)大的点要用主席树

没了

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<map>
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define mp(a,b) make_pair<int,int>(a,b)
#define cls(s) memset(s,0,sizeof(s))
#define cp pair<int,int>
#define LL long long int
using namespace std;
const int maxn = 100005,maxm = 6000005,INF = 1000000000;
inline LL read(){
LL out = 0,flag = 1; char c = getchar();
while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
return out * flag;
}
LL S[maxn],R[maxn];
int n,m,Q,bin[50];
int h[maxn],ne = 1;
struct EDGE{int to,nxt;}ed[maxn << 1];
inline void build(int u,int v){
ed[++ne] = (EDGE){v,h[u]}; h[u] = ne;
ed[++ne] = (EDGE){u,h[v]}; h[v] = ne;
}
int sum[maxm],ls[maxm],rs[maxm],rt[maxn],tot;
void modify(int& u,int pre,int l,int r,int pos){
sum[u = ++tot] = sum[pre] + 1;
ls[u] = ls[pre]; rs[u] = rs[pre];
if (l == r) return;
int mid = l + r >> 1;
if (mid >= pos) modify(ls[u],ls[pre],l,mid,pos);
else modify(rs[u],rs[pre],mid + 1,r,pos);
}
int query(int u,int v,int l,int r,LL k){
if (l == r) return l;
int mid = l + r >> 1,t = sum[ls[u]] - sum[ls[v]];
if (t >= k) return query(ls[u],ls[v],l,mid,k);
return query(rs[u],rs[v],mid + 1,r,k - t);
}
int dfn[maxn],siz[maxn],dep[maxn],Fa[maxn][18],cnt;
void dfs(int u){
dfn[u] = ++cnt; siz[u] = 1;
modify(rt[cnt],rt[cnt - 1],1,n,u);
REP(i,17) Fa[u][i] = Fa[Fa[u][i - 1]][i - 1];
Redge(u) if ((to = ed[k].to) != Fa[u][0]){
Fa[to][0] = u; dep[to] = dep[u] + 1;
dfs(to);
siz[u] += siz[to];
}
}
int hh[maxn],nne = 1,Nxt[maxn << 1],To[maxn << 1];
int fa[maxn][18],N,lk[maxn],Dep[maxn];
LL d[maxn][18];
void DFS(int u){
for (int k = hh[u]; k; k = Nxt[k]){
Dep[To[k]] = Dep[u] + 1;
DFS(To[k]);
}
}
void Build(){
LL u,b,x,r,v;
S[1] = n; R[1] = 1; N = 1;
for (int i = 1; i <= m; i++){
u = read(); v = read();
b = lower_bound(S + 1,S + 1 + N,v) - S; r = R[b];
x = query(rt[dfn[r] + siz[r] - 1],rt[dfn[r] - 1],1,n,v - S[b - 1]);
N++;
fa[N][0] = b; d[N][0] = 1 + dep[x] - dep[r];
S[N] = S[N - 1] + siz[u]; R[N] = u; lk[N] = x;
nne++;
Nxt[nne] = hh[b]; To[nne] = N; hh[b] = nne;
}
DFS(1);
//REP(i,N) printf("block %d rt = %lld lk = %d d = %lld Dep = %d total = %lld\n",i,R[i],lk[i],d[i][0],Dep[i],S[i]);
REP(j,17) REP(i,N){
fa[i][j] = fa[fa[i][j - 1]][j - 1];
d[i][j] = d[i][j - 1] + d[fa[i][j - 1]][j - 1];
}
}
LL dis(int u,int v){
if (dep[u] < dep[v]) swap(u,v);
LL re = 0;
for (int i = 0,D = dep[u] - dep[v]; bin[i] <= D; i++)
if (D & bin[i]) re += bin[i],u = Fa[u][i];
if (u == v) return re;
for (int i = 17; ~i; i--)
if (Fa[u][i] != Fa[v][i]){
u = Fa[u][i];
v = Fa[v][i];
re += bin[i + 1];
}
return re + 2;
}
LL Dis(LL a,LL b,LL x,LL y){
if (Dep[a] < Dep[b]){
swap(a,b);
swap(x,y);
}
LL re = dep[x] - dep[R[a]] + dep[y] - dep[R[b]];
int D = Dep[a] - Dep[b],u = a,v = b;
for (int i = 0; bin[i] <= D; i++)
if (D & bin[i]){
re += d[u][i];
u = fa[u][i];
}
if (u == v){
D = Dep[a] - Dep[b] - 1; u = a;
re = dep[x] - dep[R[a]];
for (int i = 0; bin[i] <= D; i++)
if (D & bin[i]){
re += d[u][i];
u = fa[u][i];
}
re += 1;
x = lk[u];
re += dis(x,y);
}
else {
for (int i = 17; ~i; i--)
if (fa[u][i] != fa[v][i]){
re += d[u][i] + d[v][i];
u = fa[u][i];
v = fa[v][i];
}
re += 2;
x = lk[u]; y = lk[v];
re += dis(x,y);
}
return re;
}
void solve(){
LL u,v,a,b,x,y;
while (Q--){
u = read(); v = read();
a = lower_bound(S + 1,S + 1 + N,u) - S;
b = lower_bound(S + 1,S + 1 + N,v) - S;
x = query(rt[dfn[R[a]] + siz[R[a]] - 1],rt[dfn[R[a]] - 1],1,n,u - S[a - 1]);
y = query(rt[dfn[R[b]] + siz[R[b]] - 1],rt[dfn[R[b]] - 1],1,n,v - S[b - 1]);
//printf("(%lld,%lld) (%lld,%lld)\n",a,x,b,y);
if (a == b) printf("%lld\n",dis(x,y));
else printf("%lld\n",Dis(a,b,x,y));
}
}
int main(){
bin[0] = 1; for (int i = 1; i <= 25; i++) bin[i] = bin[i - 1] << 1;
n = read(); m = read(); Q = read();
for (int i = 1; i < n; i++) build(read(),read());
dfs(1);
Build();
solve();
return 0;
}

BZOJ4539 [Hnoi2016]树 【倍增 + 主席树】的更多相关文章

  1. POJ 2104 K-th Number(分桶,线段树,主席树)

    一道比较经典的数据结构题.可以用多种方式来做. 一,分桶法(平方分解). 根据数字x的大小和区间内不大于x的数字数量cnt的单调性,可知第k大数kth对应的cnt应该满足cnt≥k, 且kth是满足条 ...

  2. poj 2104 K-th Number 划分树,主席树讲解

    K-th Number Input The first line of the input file contains n --- the size of the array, and m --- t ...

  3. 归并树 划分树 可持久化线段树(主席树) 入门题 hdu 2665

    如果题目给出1e5的数据范围,,以前只会用n*log(n)的方法去想 今天学了一下两三种n*n*log(n)的数据结构 他们就是大名鼎鼎的 归并树 划分树 主席树,,,, 首先来说两个问题,,区间第k ...

  4. BZOJ_2588_Spoj 10628. Count on a tree_树剖+主席树

    BZOJ_2588_Spoj 10628. Count on a tree_树剖+主席树 题意: 给定一棵N个节点的树,每个点有一个权值,对于M个询问(u,v,k),你需要回答u xor lastan ...

  5. 【BZOJ4771】七彩树(主席树)

    [BZOJ4771]七彩树(主席树) 题面 BZOJ 题解 如果没有深度限制,每次只询问子树内的颜色个数,除了树套树\(dfs\)序加前驱或者后继强行二维数点之外,还有这样一种做法: 把所有相同颜色的 ...

  6. POJ 2761 Feed the dogs(平衡树or划分树or主席树)

    Description Wind loves pretty dogs very much, and she has n pet dogs. So Jiajia has to feed the dogs ...

  7. 【题解】BZOJ3489 A Hard RMQ problem(主席树套主席树)

    [题解]A simple RMQ problem 占坑,免得咕咕咕了,争取在2h内写出代码 upd:由于博主太菜而且硬是要用指针写两个主席树,所以延后2hQAQ upd:由于博主太菜而且太懒所以他决定 ...

  8. 2019.03.25 bzoj4539: [Hnoi2016]树(主席树+倍增)

    传送门 题意:给一棵大树,令一棵模板树与这棵树相同,然后进行mmm次操作,每次选择模板树中的一个节点aaa和大树中一个节点bbb,把aaa这棵子树接在bbb上面,节点编号顺序跟aaa中的编号顺序相同. ...

  9. BNUOJ 52317 As Easy As Possible 树上倍增/主席树

    题目链接: https://acm.bnu.edu.cn/v3/problem_show.php?pid=52317 As Easy As Possible Case Time Limit: 1000 ...

随机推荐

  1. nuget在jenkins上不能自动还原项目依赖包---笔记

    最近遇到一个情况,IDE 是 VS2015 Update3 ,新建一个library项目(暂时叫做 mytests),然后用 nuget 安装了一个 Shouldly 包 在 VS 上一切正常,可以跑 ...

  2. java 创建具有参数化类型的数组

    1. List<String>[] ls; Object[] objects = ls; objects[1] = new ArrayList<Integer>(); 先把数组 ...

  3. svn清理报错:Cleanup failed to process the following paths

    这里碰到svn更新时,提示清理,清理时报错: 只需进行以下几个步骤即可解决:(原理即是清除掉svn数据库里的lock记录) 1.下载SQLiteManager,svn用的是sqlite数据库,需要一款 ...

  4. POJ 3256 (简单的DFS)

    //题意是 K N, M; //有K个牛 N个牧场,M条路 ,有向的  //把K个牛放到任意的n个不同牧场中,问所有牛都可以到达的牧场数的总和  //这是一道简单的DFS题 //k 100 //n 1 ...

  5. 【JSON类】使用说明

    理解键名路径 键名路径(keyPath)用于定位json的键,比如:{book: {title:”中国人”} },键名路径 book.title 表定位到book下的title键. 对于值是数组类型的 ...

  6. 爬虫1.1-基础知识+requests库

    目录 爬虫-基础知识+requests库 1. 状态返回码 2. URL各个字段解释 2. requests库 3. requests库爬虫的基本流程 爬虫-基础知识+requests库 关于html ...

  7. HPUX 11.31 MC/SG恢复丢失的锁盘

    有时候由于一些特殊的原因,用户的cluster中的锁盘信息丢失,或者需要更换锁盘,只要执行一个命令就可以了. #cmdisklock reset /dev/vglock:/dev/disk/diskX ...

  8. Sharepoint 2013与Sharepoint 2016的功能对比

    开发人员功能 SharePoint Foundation 2013 SharePoint Server 2013 Standard CAL SharePoint Server 2013 Enterpr ...

  9. [2018 ACL Short and System] 对话系统

    Short Paper(s) 1.  Task-oriented Dialogue System for Automatic Diagnosis. (Cited by 0) Zhongyu Wei, ...

  10. [c++] Getting Started - CMake

    CMake is an open-source cross platform build system, according to CMake's creator, Kitware. But CMak ...