题链:
https://www.luogu.org/problemnew/show/P3242
题解:
整体二分,扫描线+树状数组。
详细的题解:http://blog.csdn.net/thy_asdf/article/details/50363672
得到各个盘子影响的矩形区域后,
那么我们就是要对每个询问代表的点查询覆盖了它的权值第k小的的那个矩形。
首先有一个简化版的问题,
就是查询改点被覆盖了多少次。可以用扫描线+树状数组做。
然后对于现在询问的第k小权值,
我们就可以二分答案,即二分一个权值mid,
然后判断是否这些权值小于mid的矩形可以覆盖该点至少k次,
使得话,缩小r,否则扩大l范围。
由于有多个询问,且分别做二分的话过程是一样的,所以可以整体二分。
另外,还可以对当前扫描线上的东西建立树套树,
外面是权值线段树,里面套的是区间线段树(方便区间修改,单点查询),也可以求出第k小的权值。
如果想看具体实现的话,推荐服用这位博主的代码:http://blog.csdn.net/make_it_for_good/article/details/52985365
再另外,好像还可以树上莫队+平衡树做,但是我太弱了,忘了树上莫队,就没写了。
但是我感觉如果用莫队的话,如果数据故意把水果和盘子的左端点放到同一个位置附近,
同时程序的分块又恰好把这些位置分在了同一个块,
可能会被卡成n^2的。
代码:
#include<bits/stdc++.h> #define MAXN 160008 #define rint register int using namespace std; int N,P,Q,snt; int ANS[MAXN],fa[MAXN][18],deep[MAXN],be[MAXN],en[MAXN]; struct Edge{ int ent; int to[MAXN],nxt[MAXN],head[MAXN]; Edge(){ent=2;} void Adde(int u,int v){ to[ent]=v; nxt[ent]=head[u]; head[u]=ent++; } }E; struct Plate{ int x1,x2,y1,y2,val; }S[MAXN]; struct info{ int x,yl,yr,val,id; }A[MAXN],T[MAXN]; bool cmp1(const Plate &_A,const Plate &_B){ return _A.val<_B.val; } bool cmp2(const info &_A,const info &_B){ return _A.x<_B.x||(_A.x==_B.x&&_A.id<_B.id); } struct BIT{ int val[MAXN],n; void Reset(int _n){n=_n;} int Lowbit(int x){return x&(-x);} void Modify(int l,int r,int x){//区间修改 if(l>r) return; for(rint i=l;i<=n;i+=Lowbit(i)) val[i]+=x; for(rint i=r+1;i<=n;i+=Lowbit(i)) val[i]-=x; } int Query(int p,int ret=0){//查询覆盖次数 for(rint i=p;i>=1;i-=Lowbit(i)) ret+=val[i]; return ret; } }DT; void dfs(int u,int dad){ static int cnt; ++cnt; be[u]=cnt; fa[u][0]=dad; for(int k=1;k<18;k++) fa[u][k]=fa[fa[u][k-1]][k-1]; for(int e=E.head[u];e;e=E.nxt[e]){ int v=E.to[e]; if(v==dad) continue; deep[v]=deep[u]+1; dfs(v,u); } en[u]=cnt; } int jump(int x,int h){ for(int k=17;k>=0;k--) if(h>=(1<<k)) x=fa[x][k],h-=(1<<k); return x; } info inssegment(const Plate &rtm,int k){ static int x1,x2,y1,y2,v; x1=rtm.x1; x2=rtm.x2; y1=rtm.y1; y2=rtm.y2; v=rtm.val; assert(x1<=x2); if(k==1) return (info){x1,y1,y2,v,-1}; else return (info){x2+1,y1,y2,v,-2}; } void solve(int sl,int sr,int ql,int qr){ static int sum[MAXN]; if(ql>qr) return; if(sl==sr){ for(int i=ql;i<=qr;i++) ANS[A[i].id]=S[sl].val; return; } int mid=(sl+sr)>>1,tnt=0,qlnt=ql-1,qrnt=0; for(int i=sl;i<=mid;i++) T[++tnt]=inssegment(S[i],1),T[++tnt]=inssegment(S[i],2); for(int i=ql;i<=qr;i++) T[++tnt]=A[i]; sort(T+1,T+tnt+1,cmp2); for(int i=1;i<=tnt;i++){ if(T[i].id<0) DT.Modify(T[i].yl,T[i].yr,T[i].id==-1?1:-1); else sum[T[i].id]=DT.Query(T[i].yl); } for(int i=1;i<=tnt;i++){ if(T[i].id<0) DT.Modify(T[i].yl,T[i].yr,T[i].id==-1?-1:1); else{ if(sum[T[i].id]>=T[i].val) A[++qlnt]=T[i]; else T[i].val-=sum[T[i].id],T[++qrnt]=T[i]; } } for(int i=1;i<=qrnt;i++) A[i+qlnt]=T[i]; solve(sl,mid,ql,qlnt); solve(mid+1,sr,qlnt+1,qr); } int main(){ ios::sync_with_stdio(0); cin>>N>>P>>Q; for(int i=1,a,b;i<N;i++) cin>>a>>b,E.Adde(a,b),E.Adde(b,a); dfs(1,0); DT.Reset(N); for(int i=1,x,y,v,t;i<=P;i++){ cin>>x>>y>>v; if(be[x]>be[y]) swap(x,y); if(be[x]<=be[y]&&en[x]>=en[y]){ t=jump(y,deep[y]-deep[x]-1); S[++snt]=(Plate){1,be[t]-1,be[y],en[y],v}; if(en[t]+1<=N) S[++snt]=(Plate){be[y],en[y],en[t]+1,N,v}; } else S[++snt]=(Plate){be[x],en[x],be[y],en[y],v}; } sort(S+1,S+snt+1,cmp1); for(int i=1,x,y,k;i<=Q;i++){ cin>>x>>y>>k; if(be[x]>be[y]) swap(x,y); A[i]=(info){be[x],be[y],0,k,i}; //在构建的二维平面里,x点对应的下标不是x,而是be[x] //wa了一次,就是因为上面的那行代码写成了: //A[i]=(info){x,y,0,k,i}; } solve(1,snt,1,Q); for(int i=1;i<=Q;i++) cout<<ANS[i]<<endl; return 0; }