hdu 5840 This world need more Zhu (2016CCPC 网络赛1009) 分块+线段树

很明显分块，但是如果k>block我们不能用倍增，那怎么做？很简单dfs记录队列中的顺序即可。然后k<=block的用线段树维护一下，也就是dfs的时候用深度维护一下。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<cmath>
#define mp(a,b) make_pair(a,b)
using namespace std;
const int maxn=410005;
const int N=100005;
int blo;
struct pi{
    int ma;
}pp[11][maxn];
int pre[501],no[501];
int val[N];
void build(int pre,int tot,int l,int r){
    pp[pre][tot].ma=0;
    if(l==r) return;
    build(pre,tot<<1,l,(l+r)>>1);
    build(pre,(tot<<1)|1,((l+r)>>1)+1,r);
}
void merg(int pre,int tot,int le,int ri,int x,int q){
    if(le==ri){
        pp[pre][tot].ma=q;
        return;
    }
    int mid=(le+ri)>>1;
    if(x<=mid) merg(pre,tot<<1,le,mid,x,q);
    else merg(pre,(tot<<1)|1,mid+1,ri,x,q);
    pp[pre][tot].ma=max(pp[pre][(tot<<1)].ma,pp[pre][((tot<<1)|1)].ma);
}
int query(int pre,int tot,int le,int ri,int l,int r){
    if(l>r) return 0;
    if(l<=le&&ri<=r){
        return pp[pre][tot].ma;
    }
    int s=0;
    int mid=(le+ri)>>1;
    if(l<=mid) s=max(s,query(pre,tot<<1,le,mid,l,r));
    if(r>mid)  s=max(s,query(pre,(tot<<1)|1,mid+1,ri,l,r));
    return s;
}
int deap[N],kk[19][N];
int maxlog;
vector<int>g[N];
void dfs(int v,int p,int d)
{
    deap[v]=d;
    int i,k;
    kk[0][v]=p;
    for(i=1;i<maxlog;i++){
        if(kk[i-1][v]<0)kk[i][v]=-1;
        else{
            kk[i][v]=kk[i-1][kk[i-1][v]];
        }
    }
    k=g[v].size();
    for(i=0;i<k;i++){
        if(g[v][i]!=p){
            dfs(g[v][i],v,d+1);
        }
    }
    return ;
}
int find(int a,int b){
    if(deap[a]>deap[b])swap(a,b);
    int i,f;
    f=deap[b]-deap[a];
    for(i=0;i<maxlog;i++){
        if((f>>i)&1)
            b=kk[i][b];
    }
    if(b==a)
        return a;
    for(i=maxlog-1;i>=0;i--)
    {
        if(kk[i][a]!=kk[i][b])
        {
            a=kk[i][a];
            b=kk[i][b];
        }
    }
    return kk[0][a];
}
int ans[N];
typedef  pair<int,int>p1;
typedef pair<p1,int>p2;
typedef pair<p1,p1>p3;
vector<p1>gg[N];
template <class T>
inline void scan_d(T &ret) {
    char c; ret=0;
    while((c=getchar())<'0'||c>'9');
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
}
vector<p2>g2[501];
void dfs1(int u,int pa,int k){
    int u1=(deap[u]-1)%k;
    u1++;
    int y=(deap[u]-u1)/k+1;
    merg(u1,1,1,no[u1],y,val[u]);
    for(p1 v:gg[u]){
        int w=(deap[v.first]-1)%k;
        w++;
        int x;
        if(w<=u1) x=(deap[v.first]-w)/k+1;
        else x=(deap[v.first]-w)/k+2;
        ans[v.second]=max(ans[v.second],query(u1,1,1,no[u1],x,y));
    }
    for(int v:g[u]){
        if(v!=pa){
            dfs1(v,u,k);
        }
    }
    merg(u1,1,1,no[u1],y,0);
}
vector<p3>g3[N];
int a[N];
int tot;
void dfs3(int u,int pa){
    a[++tot]=u;
    for(p3 v:g3[u]){
        int no=tot-v.first.second;
        while(no>0){
            if(deap[a[no]]<deap[v.first.first]) break;
            ans[v.second.second]=max(ans[v.second.second],val[a[no]]);
            no=no-v.second.first;
        }
    }
    for(int v:g[u]){
        if(v!=pa){
            dfs3(v,u);
        }
    }
    tot--;
}
int main()
{
    int t,NN=0;
    cin>>t;
    maxlog=19;
    while (t--) {
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) scanf("%d",&val[i]);
        blo=2;
        for(int i=1;i<=n;i++){
            g[i].clear();
            gg[i].clear();
            g3[i].clear();
        }
        for(int i=1;i<500;i++) g2[i].clear();
        for(int i=1;i<n;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            g[x].push_back(y);
            g[y].push_back(x);
        }
        dfs(1,-1,1);
        for(int i=0;i<m;i++){
            int u,v,k;
            scanf("%d%d%d",&u,&v,&k);
            ans[i]=0;
            if(k>n){
                continue;
            }
            if(k>blo){
                int w=find(u,v);
                if(w==u&&w==v){
                    continue;
                }
                if(w!=u){
                    g3[u].push_back(mp(mp(w,k-1),mp(k,i)));
                }
                if(w!=v){
                    int q=deap[u]+deap[v]-2*deap[w]+1;
                    q=q%k;
                    g3[v].push_back(mp(mp(w,q),mp(k,i)));
                }
                ans[i]=0;
            }
            else{
                g2[k].push_back(mp(mp(u,v),i));
                ans[i]=0;
            }
        }
        tot=0;
        dfs3(1,-1);
        for(int i=1;i<=blo;i++){
            for(int j=1;j<=i;j++){
                no[j]=n/i+2;
                build(j,1,1,no[j]);
            }
            for(p2 v1:g2[i]){
                int w=find(v1.first.first,v1.first.second);
                if(w!=v1.first.second){
                    int u=v1.first.first;
                    int v=v1.first.second;
                    int q=deap[u]+deap[v]-2*deap[w]+1;
                    q=q%i;
                    int no=0;
                    while(v!=-1&&q){
                        if(q&1){
                            v=kk[no][v];
                        }
                        no++;
                        q>>=1;
                    }
                    if(v!=-1&&deap[v]>=deap[w]){
                        gg[v].push_back(mp(w,v1.second));
                    }
                }
                if(w!=v1.first.first){
                    int u=v1.first.first;
                    int no=0;
                    int q=i-1;
                    while(u!=-1&&q){
                        if(q&1){
                            u=kk[no][u];
                        }
                        no++;
                        q>>=1;
                    }
                    if(u!=-1&&deap[u]>=deap[w]){
                        gg[u].push_back(mp(w,v1.second));
                    }
                }
                if(v1.first.first==v1.first.second){
                    if(i==1){
                        ans[v1.second]=val[v1.first.first];
                    }
                }
            }
            dfs1(1,-1,i);
        }
        printf("Case #%d:\n",++NN);
        for(int i=0;i<m;i++){
            printf("%d\n",ans[i]);
        }
    }
}