【GDOI2017模拟11.5】Dash Speed

Description

给出n个点的一棵树，每条边有一个限制区间[l,r]。
给出m次询问，每次询问给出一个数x，求一条最长的所有边都满足l<=x<=r的路径。
n,m<=7*10^4

Solution

一开始看错数据范围，以为是7*10^5，虚的我。。。
然而没想到每次询问就是相当于保留所有l<=x<=r的边，求森林的直径
考虑分治（据说这也是cdq分治）
首先把每一条边放到所有它完全包含的分治区间中。
然后当我们做到当前区间时，把这个区间中的所有边加入森林中去。
回溯时候再把这些边删掉。
不过因为要兹瓷删边，所以我们的并查集不能路径压缩，只能按秩合并。。。
跑的慢死了。。。

Code

#include <cmath>#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])using namespace std;const int N=7*1e4+5;struct note{int a,pa,pb,x,bz;}pr[N*20];int n,m,cnt,tot,ans,x,y,l,r,mx,mi[18],ln[N*2];int d[N],f[N],fir[N],dfn[N*2],g[N*2][18],p[N][2],q[N*20],rank[N],an[N];int t[N*2],next[N*2],last[N],la[N*4],ne[N*20],A[N*20],B[N*20];void add(int x,int y) {    t[++cnt]=y;next[cnt]=last[x];last[x]=cnt;}int max(int x,int y) {    if (x>y) return x;else return y;}void insert(int v,int l,int r,int x,int y,int a,int b) {    if (x<=l&&r<=y) {        A[++tot]=a;B[tot]=b;mx=max(mx,v);        ne[tot]=la[v];la[v]=tot;        return;    }    int m=(l+r)/2;    if (x<=m) insert(v*2,l,m,x,y,a,b);    if (y>m) insert(v*2+1,m+1,r,x,y,a,b);}void dfs(int x,int y) {    d[x]=d[y]+1;dfn[++tot]=x;fir[x]=tot;    rep(i,x) if (t[i]!=y) dfs(t[i],x),dfn[++tot]=x;}int get(int x) {    return f[x]?get(f[x]):x;}int lca(int x,int y) {    x=fir[x];y=fir[y];    if (x>y) swap(x,y);    int z=ln[y-x+1];    if (d[dfn[g[x][z]]]<d[dfn[g[y-mi[z]+1][z]]]) return dfn[g[x][z]];    else return dfn[g[y-mi[z]+1][z]];}int len(int x,int y) {    int z=lca(x,y);    return d[x]+d[y]-2*d[z];}void merge(int x,int &ans) {    int ax=get(A[x]),ay=get(B[x]);tot++;    int now[2],cnt;now[0]=p[ax][0];now[1]=p[ax][1];cnt=len(now[0],now[1]);    int dist=len(p[ay][0],p[ay][1]);    if (dist>cnt) cnt=dist,now[0]=p[ay][0],now[1]=p[ay][1];    fo(j,0,1) fo(k,0,1) {        dist=len(p[ax][j],p[ay][k]);        if (dist>cnt) cnt=dist,now[0]=p[ax][j],now[1]=p[ay][k];    }    if (rank[ay]<rank[ax]) {        pr[x].x=ay;pr[x].a=ax;        pr[x].pa=p[ax][0];pr[x].pb=p[ax][1];        p[ax][0]=now[0];p[ax][1]=now[1];        f[ay]=ax;    } else {        pr[x].x=ax;pr[x].a=ay;        pr[x].pa=p[ay][0];pr[x].pb=p[ay][1];        p[ay][0]=now[0];p[ay][1]=now[1];        f[ax]=ay;        if (rank[ax]==rank[ay]) rank[ay]++,pr[x].bz=1;    }    ans=max(ans,len(now[0],now[1]));}void del(int x) {    for(int i=la[x];i;i=ne[i]) q[++q[0]]=i;    while (q[0]) {        p[pr[q[q[0]]].a][0]=pr[q[q[0]]].pa;        p[pr[q[q[0]]].a][1]=pr[q[q[0]]].pb;        f[pr[q[q[0]]].x]=0;rank[pr[q[q[0]]].a]-=pr[q[q[0]]].bz;        q[0]--;    }}void work(int v,int l,int r,int ans) {    for(int i=la[v];i;i=ne[i]) merge(i,ans);    if (l==r) {        an[l]=ans;        del(v);        return;    }    int m=(l+r)/2;    work(v*2,l,m,ans);    work(v*2+1,m+1,r,ans);    del(v);}int main() {    freopen("speed.in","r",stdin);    freopen("speed.out","w",stdout);    scanf("%d%d",&n,&m);    fo(i,1,n-1) {        scanf("%d%d%d%d",&x,&y,&l,&r);        add(x,y);add(y,x);        insert(1,1,n,l,r,x,y);    }    tot=0;dfs(1,0);mi[0]=1;    fo(i,1,tot) g[i][0]=i,ln[i]=log(i)/log(2);fo(i,1,17) mi[i]=mi[i-1]*2;    fo(j,1,17) fo(i,1,tot-mi[j]+1)         if (d[dfn[g[i][j-1]]]<d[dfn[g[i+mi[j-1]][j-1]]]) g[i][j]=g[i][j-1];        else g[i][j]=g[i+mi[j-1]][j-1];    fo(i,1,n) p[i][0]=p[i][1]=i;    work(1,1,n,0);    fo(i,1,m) scanf("%d",&x),printf("%d\n",an[x]);}