接水果(fruit)——整体二分+扫描线
题目
【题目描述】
风见幽香非常喜欢玩一个叫做 osu! 的游戏，其中她最喜欢玩的模式就是接水果。由于她已经 DT FC 了 The big black，她觉得这个游戏太简单了，于是发明了一个更加难的版本。
首先有一个地图，是一棵由 $n$ 个顶点、$n-1$ 条边组成的树（例如图 $1$ 给出的树包含 $8$ 个顶点、$7$ 条边）。这颗树上有 P 个盘子，每个盘子实际上是一条路径（例如图 $1$ 中顶点 $6$ 到顶点 $8$ 的路径），并且每个盘子还有一个权值。第 $i$ 个盘子就是顶点 $a_i$ 到顶点 $b_i$ 的路径(由于是树，所以从 $a_i$ 到 $b_i$ 的路径是唯一的)，权值为 $c_i$。接下来依次会有 $Q$ 个水果掉下来，每个水果本质上也是一条路径，第 $i$ 个水果是从顶点 $u_i$ 到顶点 $v_i$ 的路径。
幽香每次需要选择一个盘子去接当前的水果：一个盘子能接住一个水果，当且仅当盘子的路径是水果的路径的子路径（例如图 $1$ 中从 $3$ 到 $7$ 的路径是从 $1$ 到 $8$ 的路径的子路径）。这里规定：从 $a$ 到 $b$ 的路径与从 $b$ 到 $a$ 的路径是同一条路径。当然为了提高难度，对于第 $i$ 个水果，你需要选择能接住它的所有盘子中，权值第 $k_i$ 小的那个盘子，每个盘子可重复使用（没有使用次数的上限：一个盘子接完一个水果后，后面还可继续接其他水果，只要它是水果路径的子路径）。幽香认为这个游戏很难，你能轻松解决给她看吗？
【输入格式】
第一行三个数 $n$ 和 $P$ 和 $Q$，表示树的大小和盘子的个数和水果的个数。
接下来 $n-1$ 行，每行两个数 $a$、$b$，表示树上的 $a$ 和 $b$ 之间有一条边。树中顶点按 $1$ 到 $n$ 标号。 接下来 $P$ 行，每行三个数 $a$、$b$、$c$，表示路径为 $a$ 到 $b$、权值为 $c$ 的盘子，其中 $0 \leq c \leq 10^9, \ a \neq b$。
接下来 $Q$ 行，每行三个数 $u$、$v$、$k$，表示路径为 $u$ 到 $v$ 的水果，其中 $u$ 不等于 $v$，你需要选择第 $k$ 小的盘子，第 $k$ 小一定存在。
【输出格式】
对于每个果子，输出一行表示选择的盘子的权值。
【样例输入】
10 10 10
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
3 2 217394434
10 7 13022269
6 7 283254485
6 8 333042360
4 6 442139372
8 3 225045590
10 4 922205209
10 8 808296330
9 2 486331361
4 9 551176338
1 8 5
3 8 3
3 8 4
1 8 3
4 8 1
2 3 1
2 3 1
2 3 1
2 4 1
1 4 1
【样例输出】
442139372
333042360
442139372
283254485
283254485
217394434
217394434
217394434
217394434
217394434
【数据范围与提示】
对于所有数据，$N,P,Q \leq 40000$。
题解
考虑一个盘子 $ u,v $ 能接到的水果的范围，那么水果的 $ s,t $ 必须分别属于盘子两个的两个子树，问题即转化成 $ s,t $ 属于的第 $k$ 大的盘子
因为一个子树在 dfs 序上是一个区间，一个盘子就是两个区间的包含关系，那么映射到坐标系上盘子就成了一个矩形，水果就成了一个点，问题变成覆盖某一个点的第 $k$ 大的矩形
考虑如何解决，整体二分矩形，判断覆盖个数是否大于，然后左右递归即可
至于判断，类似扫描线的思想，将矩形按一维排序，另一维区间用数据结构（树状数组）维护，单点查询即可
如果水果 $ u,v $ 的 lca 为 $ u $ 或 $ v $ 时，需要特判一下
时间效率：$ O(n \log^2 n) $
代码
1 #include<bits/stdc++.h> 2 const int N=8e4+10; 3 using namespace std; 4 int n,m,q,tot,fa[N][18],dfn[N],tim,last[N],dep[N],ans[N],sum[N],head[N],cnt; 5 struct edge{int to,nex;}e[N<<1]; 6 struct Plate{int x1,x2,y1,y2,v;}pla[N]; 7 struct node{int x,y1,y2,v,id;}eve[N]; 8 struct fruit{int x,y,k,id;}fru[N],s1[N],s2[N]; 9 bool cmp(Plate a,Plate b){return a.v<b.v;} 10 bool cmp1(node a,node b){return a.x==b.x?a.id<b.id:a.x<b.x;} 11 void add(int s,int t){e[++cnt]=(edge){t,head[s]},head[s]=cnt;} 12 13 class Bit{ 14 public: 15 int val[N]; 16 void modify(int l,int r,int v){ 17 for (int i=l;i<=n;i+=(i&(-i))) val[i]+=v; 18 for (int i=r+1;i<=n;i+=(i&(-i))) val[i]-=v; 19 } 20 int query(int x){int res=0;for (;x;x-=(x&(-x))) res+=val[x];return res;} 21 }T; 22 23 void dfs(int x){ 24 dfn[x]=++tim; 25 for (int i=0;fa[x][i];i++) fa[x][i+1]=fa[fa[x][i]][i]; 26 for (int k=head[x],v;k;k=e[k].nex) 27 if ((v=e[k].to)!=fa[x][0]) 28 fa[v][0]=x,dep[v]=dep[x]+1,dfs(v); 29 last[x]=tim; 30 } 31 int jump(int a,int h){for (int i=16;h;i--) if (h>=(1<<i)) h-=(1<<i),a=fa[a][i];return a;} 32 int lca(int a,int b){ 33 if (dep[a]<dep[b]) swap(a,b); 34 a=jump(a,dep[a]-dep[b]); 35 if (a==b) return a; 36 for (int i=16;i>=0;i--) if (fa[a][i]!=fa[b][i]) a=fa[a][i],b=fa[b][i]; 37 return fa[a][0]; 38 } 39 40 void solve(int l,int r,int st,int ed){ 41 if (st>ed) return; 42 if (l==r){ 43 for (int i=st;i<=ed;i++) ans[fru[i].id]=pla[l].v; 44 return; 45 } 46 int mid=(l+r)>>1,siz=0; 47 for (int i=l;i<=mid;i++){ 48 eve[++siz]=(node){pla[i].x1,pla[i].y1,pla[i].y2,1,0}; 49 eve[++siz]=(node){pla[i].x2,pla[i].y1,pla[i].y2,-1,n+1}; 50 } 51 for (int i=st;i<=ed;i++) eve[++siz]=(node){fru[i].x,fru[i].y,0,0,i}; 52 sort(eve+1,eve+1+siz,cmp1); 53 for (int i=1;i<=siz;i++) 54 if (st<=eve[i].id&&eve[i].id<=ed) sum[eve[i].id]=T.query(eve[i].y1); 55 else T.modify(eve[i].y1,eve[i].y2,eve[i].v); 56 int a=0,b=0; 57 for (int i=st;i<=ed;i++) { 58 if (sum[i]>=fru[i].k) s1[++a]=fru[i]; 59 else s2[++b]=(fruit){fru[i].x,fru[i].y,fru[i].k-sum[i],fru[i].id}; 60 } 61 for (int i=st;i<=st+a-1;i++) fru[i]=s1[i-st+1]; 62 for (int i=st+a;i<=ed;i++) fru[i]=s2[i-st-a+1]; 63 solve(l,mid,st,st+a-1),solve(mid+1,r,st+a,ed); 64 } 65 66 int main(){ 67 scanf("%d%d%d",&n,&m,&q); 68 for (int i=1,a,b;i<n;i++) scanf("%d%d",&a,&b),add(a,b),add(b,a); 69 dfs(1); 70 for (int i=1,a,b,c,u;i<=m;i++){ 71 scanf("%d%d%d",&a,&b,&c);u=lca(a,b); 72 if (dfn[a]>dfn[b]) swap(a,b); 73 if (u!=a) pla[++tot]=(Plate){dfn[a],last[a],dfn[b],last[b],c}; 74 else{ 75 int w=jump(b,dep[b]-dep[a]-1); 76 pla[++tot]=(Plate){1,dfn[w]-1,dfn[b],last[b],c}; 77 if (last[w]<n) pla[++tot]=(Plate){dfn[b],last[b],last[w]+1,n,c}; 78 } 79 } 80 sort(pla+1,pla+1+tot,cmp); 81 for (int i=1,a,b,k;i<=q;i++){ 82 scanf("%d%d%d",&a,&b,&k); 83 if (dfn[a]>dfn[b]) swap(a,b); 84 fru[i]=(fruit){dfn[a],dfn[b],k,i}; 85 } 86 solve(1,tot,1,q); 87 for (int i=1;i<=q;i++) printf("%d\n",ans[i]); 88 return 0; 89 }
