【HNOI2016模拟4.13】a

Description

给出一棵n个节点的树，每个节点ai的范围在1~m，和q次询问，每次询问x到y的路径中小于/等于/大于k的数的个数。强制在线。

n,q<=300000,m<=150000

Solution

裸题，用来练模板。

有两种做法，一种是主席树用lca直接搞。

一种是链剖开m棵线段树暴力搞。

这里打的是第一种。

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define rep(i,a) for(int i=last[a];i;i=next[i])
#define N 270000
using namespace std;
struct note{int sum,l,r;}tr[N*30];
int n,m,q,l,tot,root
,a
,d
,f
[20],ansl,ans,ansr,x,y,z;
int t[N*2],next[N*2],last
;
void add(int x,int y) {
t[++l]=y;next[l]=last[x];last[x]=l;
}
void insert(int &v,int l,int r,int x) {
tr[++tot]=tr[v];tr[tot].sum++;v=tot;
if (l==r) return;
int m=(l+r)/2;
if (x<=m) insert(tr[v].l,l,m,x);
else insert(tr[v].r,m+1,r,x);
}
void dfs(int x,int y) {
root[x]=root[y];f[x][0]=y;d[x]=d[y]+1;
insert(root[x],1,m,a[x]);
rep(i,x) if (t[i]!=y) dfs(t[i],x);
}
int lca(int x,int y) {
if (d[x]<d[y]) swap(x,y);
fd(j,19,0) if (d[f[x][j]]>d[y]) x=f[x][j];
if (d[x]!=d[y]) x=f[x][0];
fd(j,19,0) if (f[x][j]!=f[y][j]) x=f[x][j],y=f[y][j];
if (x!=y) return f[x][0];else return x;
}
int find(int v,int l,int r,int x,int y) {
if (!v) return 0;
if (l==x&&r==y) return tr[v].sum;
int m=(l+r)/2;
if (y<=m) return find(tr[v].l,l,m,x,y);
else if (x>m) return find(tr[v].r,m+1,r,x,y);
else return find(tr[v].l,l,m,x,m)+find(tr[v].r,m+1,r,m+1,y);
}
int query(int x,int y,int l,int r) {
if (l>r) return 0;
int z=lca(x,y);
return find(root[x],1,m,l,r)+find(root[y],1,m,l,r)-
find(root[z],1,m,l,r)-find(root[f[z][0]],1,m,l,r);
}
int main() {
scanf("%d%d%d",&n,&m,&q);
fo(i,1,n) scanf("%d",&a[i]);
fo(i,1,n-1) scanf("%d%d",&x,&y),add(x,y),add(y,x);
dfs(1,0);
fo(j,1,19) fo(i,1,n) f[i][j]=f[f[i][j-1]][j-1];
for(;q;q--) {
scanf("%d%d%d",&x,&y,&z);int lastans=ansl^ans^ansr;
x^=lastans;y^=lastans;z^=lastans;
ansl=query(x,y,1,z-1);ans=query(x,y,z,z);ansr=query(x,y,z+1,m);
printf("%d %d %d\n",ansl,ans,ansr);
}
}