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
;
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
,kk[19]
;
int maxlog;
vector<int>g
;
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
;
typedef pair<int,int>p1;
typedef pair<p1,int>p2;
typedef pair<p1,p1>p3;
vector<p1>gg
;
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
;
int a
;
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]);
}
}
}