题目链接:http://www.lydsy.com:808/JudgeOnline/problem.php?id=3460
题意:一棵树。每个节点住一个人,这个人打水的时间为Ti。每次查询一个路径。这个路径上的人都去一个水管前打水。总的最小等待时间。
思路:在树的DFS序中分块。
const int N=200005;
struct node
{
int x,y,lca,flag,id;
int t;
};
vector<node> b;
vector<int> g[N];
vector<pair<int,int> > Q[N];
int n,m,key,d[N];
int S[N],id;
int mp[N],pos[N][2];
int get(int x)
{
if(S[x]!=x) S[x]=get(S[x]);
return S[x];
}
int visit[N];
void dfs(int u,int pre)
{
pos[u][0]=++id;
mp[id]=u;
int i;
for(i=0;i<SZ(g[u]);i++)
{
int v=g[u][i];
if(v==pre) continue;
dfs(v,u);
S[v]=u;
}
pos[u][1]=++id;
mp[id]=u;
visit[u]=1;
for(i=0;i<SZ(Q[u]);i++)
{
int v=Q[u][i].first;
int id=Q[u][i].second;
if(visit[v]) b[id].lca=get(v);
}
}
i64 ans[N][2];
int M;
int MM;
int cmp(node a,node b)
{
if(a.x/MM!=b.x/MM) return a.x<b.x;
if((a.x/MM)&1) return a.y>b.y;
return a.y<b.y;
}
int dp[N];
i64 A[N],B[N];
i64 suma(int x)
{
i64 ans=0;
while(x<N) ans+=A[x],x+=x&-x;
return ans;
}
i64 sumb(int x)
{
i64 ans=0;
while(x) ans+=B[x],x-=x&-x;
return ans;
}
i64 Ans;
int qp[N];
void add(int x)
{
int j;
for(j=d[mp[x]];j;j-=j&-j) A[j]+=qp[mp[x]];
if(qp[mp[x]]==-1) Ans-=sumb(d[mp[x]]);
for(j=d[mp[x]];j<=n;j+=j&-j) B[j]+=(i64)dp[d[mp[x]]]*qp[mp[x]];
Ans+=suma(d[mp[x]]+1)*dp[d[mp[x]]]*qp[mp[x]];
if(qp[mp[x]]==1) Ans+=sumb(d[mp[x]]);
qp[mp[x]]*=-1;
}
void deal()
{
MM=sqrt(2*n);
sort(b.begin(),b.end(),cmp);
int L=1,R=0;
int i;
for(i=0;i<N;i++) qp[i]=1;
for(i=0;i<SZ(b);i++)
{
int ll=b[i].x;
int rr=b[i].y;
int id=b[i].id;
int t=b[i].t;
while(R<rr) add(++R);
while(R>rr) add(R--);
while(L<ll) add(L++);
while(L>ll) add(--L);
if(b[i].flag) add(pos[b[i].lca][0]);
ans[id][t]=Ans;
if(b[i].flag) add(pos[b[i].lca][0]);
}
}
int main()
{
n=getInt();
m=getInt();
key=getInt();
int i;
for(i=1;i<=n;i++) d[i]=getInt(),dp[i]=d[i];
sort(dp+1,dp+n+1);
M=unique(dp+1,dp+n+1)-(dp+1);
for(i=1;i<=n;i++) d[i]=lower_bound(dp+1,dp+M+1,d[i])-dp;
int root;
for(i=1;i<=n;i++)
{
int x=getInt();
if(!x) root=i;
else g[x].pb(i),g[i].pb(x);
}
int cnt=0;
for(i=1;i<=m;i++)
{
char op[5];
scanf("%s",op);
if(op[0]=='C') root=getInt();
else
{
int x=getInt();
node a;
a.x=x%n+1;
a.y=root;
a.id=i;
a.t=0;
b.pb(a); Q[a.x].pb(MP(a.y,cnt)),Q[a.y].pb(MP(a.x,cnt)); cnt++;
a.x=(x+key)%n+1;
a.y=root;
a.id=i;
a.t=1;
b.pb(a); Q[a.x].pb(MP(a.y,cnt)),Q[a.y].pb(MP(a.x,cnt)); cnt++;
}
}
for(i=1;i<=n;i++) S[i]=i;
dfs(1,0);
for(i=0;i<SZ(b);i++)
{
if(b[i].y==b[i].lca) swap(b[i].x,b[i].y);
if(b[i].x==b[i].lca)
{
b[i].flag=0;
b[i].x=pos[b[i].x][0];
b[i].y=pos[b[i].y][0];
continue;
}
if(pos[b[i].x][1]>pos[b[i].y][0]) swap(b[i].x,b[i].y);
b[i].x=pos[b[i].x][1];
b[i].y=pos[b[i].y][0];
b[i].flag=1;
}
clr(ans,-1);
deal();
int last=0;
for(i=1;i<=m;i++) if(ans[i][0]!=-1||ans[i][1]!=-1)
{
printf("%lld\n",ans[i][last&1]);
last=ans[i][last&1];
}
}