[BZOJ4012][HNOI2015]开店(动态点分治)

 1 #include<cstdio>
 2 #include<vector>
 3 #include<algorithm>
 4 #include<iostream>
 5 #define rep(i,l,r) for (int i=l; i<=r; i++)
 6 #define For(i,x) for (int i=h[x],k; i; i=nxt[i])
 7 #define pb push_back
 8 typedef long long ll;
 9 using namespace std;
10 
11 const int N=200100,inf=1000000000;
12 int n,Q,A,u,v,w,L,R,l,r,cnt,tot,S,rt,vis[N],lg[N<<2],pos[N],a[N],h[N],sz[N],f[N],fa[N],to[N<<1],val[N<<1],nxt[N<<1];
13 struct D{
14     int val; ll dis,sum;
15     D(){}; D(int a,int b,int c):val(a),dis(b),sum(c){};
16     bool operator <(const D &a)const{ return val==a.val ? dis==a.dis ? sum<a.sum : dis<a.dis : val<a.val; }
17 };
18 vector<D>va[N],vb[N];
19 ll d[N],st[20][N<<2],ans;
20 
21 template<typename T>inline void rd(T &x){
22     int t; char ch;
23     for (t=0; !isdigit(ch=getchar()); t=(ch==‘-‘));
24     for (x=ch-‘0‘; isdigit(ch=getchar()); x=x*10+ch-‘0‘);
25     if (t) x=-x;
26 }
27 
28 void add(int u,int v,int w){ to[++cnt]=v; val[cnt]=w; nxt[cnt]=h[u]; h[u]=cnt; }
29 
30 void dfs(int x,int fa){
31     st[0][++tot]=d[x]; pos[x]=tot;
32     For(i,x) if ((k=to[i])!=fa) d[k]=d[x]+val[i],dfs(k,x),st[0][++tot]=d[x];
33 }
34 
35 void rmq(){ rep(i,1,18) rep(j,1,tot-(1<<i)+1) st[i][j]=min(st[i-1][j],st[i-1][j+(1<<(i-1))]); }
36 
37 ll dis(int u,int v){
38     int x=pos[u],y=pos[v];
39     if (x>y) swap(x,y);
40     int t=lg[y-x+1];
41     return d[u]+d[v]-2*min(st[t][x],st[t][y-(1<<t)+1]);
42 }
43 
44 void find(int x,int fa){
45     sz[x]=1; f[x]=0;
46     For(i,x) if ((k=to[i])!=fa && !vis[k]) find(k,x),sz[x]+=sz[k],f[x]=max(f[x],sz[k]);
47     f[x]=max(f[x],S-sz[x]);
48     if (f[x]<f[rt]) rt=x;
49 }
50 
51 void solve(int x){
52     vis[x]=1;
53     For(i,x) if (!vis[k=to[i]]) S=sz[k],f[rt=0]=inf,find(k,x),fa[rt]=x,solve(rt);
54 }
55 
56 void work(){
57     rep(x,1,n) for (int i=x; i; i=fa[i]) va[i].pb(D(a[x],dis(i,x),0)),vb[i].pb(D(a[x],dis(x,fa[i]),0));
58     rep(i,1,n){
59         va[i].pb(D(-1,0,0)); va[i].pb(D(1<<30,0,0));
60         vb[i].pb(D(-1,0,0)); vb[i].pb(D(1<<30,0,0));
61         sort(va[i].begin(),va[i].end()); sort(vb[i].begin(),vb[i].end());
62         for (int j=1; j<(int)va[i].size(); j++) va[i][j].sum=va[i][j-1].sum+va[i][j].dis;
63         for (int j=1; j<(int)vb[i].size(); j++) vb[i][j].sum=vb[i][j-1].sum+vb[i][j].dis;
64     }
65 }
66 
67 ll que(int x,int p){
68     ll ans=0; p++;
69     for (int i=x; i; i=fa[i]){
70         int t=lower_bound(va[i].begin(),va[i].end(),D(p,0,0))-va[i].begin()-1;
71         ans+=va[i][t].sum+t*dis(i,x);
72     }
73     for (int i=x; fa[i]; i=fa[i]){
74         int t=lower_bound(vb[i].begin(),vb[i].end(),D(p,0,0))-vb[i].begin()-1;
75         ans-=vb[i][t].sum+t*dis(fa[i],x);
76     }
77     return ans;
78 }
79 
80 int main(){
81     rd(n); rd(Q); rd(A);
82     lg[1]=0; rep(i,2,n<<2) lg[i]=lg[i>>1]+1;
83     rep(i,1,n) rd(a[i]);
84     rep(i,1,n-1) rd(u),rd(v),rd(w),add(u,v,w),add(v,u,w);
85     dfs(1,0); rmq(); S=n; f[rt=0]=inf; find(1,0); solve(rt); work();
86     while (Q--){
87         rd(u); rd(l); rd(r);
88         L=min((l+ans)%A,(r+ans)%A); R=max((l+ans)%A,(r+ans)%A);
89         printf("%lld\n",ans=que(u,R)-que(u,L-1));
90     }
91     return 0;
92 }