bzoj 4012: [HNOI2015]开店

Description
风见幽香有一个好朋友叫八云紫，她们经常一起看星星看月亮从诗词歌赋谈到
人生哲学。最近她们灵机一动，打算在幻想乡开一家小店来做生意赚点钱。这样的
想法当然非常好啦，但是她们也发现她们面临着一个问题，那就是店开在哪里，面
向什么样的人群。很神奇的是，幻想乡的地图是一个树形结构，幻想乡一共有 n
个地方，编号为 1 到 n，被 n-1 条带权的边连接起来。每个地方都住着一个妖怪，
其中第 i 个地方的妖怪年龄是 x_i。妖怪都是些比较喜欢安静的家伙，所以它们并
不希望和很多妖怪相邻。所以这个树所有顶点的度数都小于或等于 3。妖怪和人一
样，兴趣点随着年龄的变化自然就会变化，比如我们的 18 岁少女幽香和八云紫就
比较喜欢可爱的东西。幽香通过研究发现，基本上妖怪的兴趣只跟年龄有关，所以
幽香打算选择一个地方 u（u为编号），然后在 u开一家面向年龄在 L到R 之间（即
年龄大于等于 L、小于等于 R）的妖怪的店。也有可能 u这个地方离这些妖怪比较
远，于是幽香就想要知道所有年龄在 L 到 R 之间的妖怪，到点 u 的距离的和是多
少（妖怪到 u 的距离是该妖怪所在地方到 u 的路径上的边的权之和） ，幽香把这个
称为这个开店方案的方便值。幽香她们还没有决定要把店开在哪里，八云紫倒是准
备了很多方案，于是幽香想要知道，对于每个方案，方便值是多少呢。

解题报告:
前情:本人的只是水过.
我们知道：\(dis[u,v]=d[u]+d[v]-2*d[lca(u,v)]\)
对于前两项，我们是已知的,对于\(d[lca(u,v)]\)的求法十分有意思,其实就是u和v到1的路径的交，这样，我们就可以先用所有的v，将v到1的路径打上标记，然后统计答案时就是u到1的路径上的标记和，用树链剖分即可\(logn\)的标记和查询，然后关于年龄限制，我们只需将树链剖分的线段树可持久化一下即可.
一些注意的地方:
一开始标记打错，应该标记上记录的是经过次数而不是边权×次数，然后就是区间修改主席树的基本操作了.

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#define RG register
#define il inline
#define iter iterator
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
const int N=150005;
int head[N],num=0,nxt[N<<1],to[N<<1],dis[N<<1],n,Q,A,edg[N],tot,val[N];
int bt=0,b[N],a[N];ll sum[N],w[N],sed[N];
int gi(){
    int str=0;char ch=getchar();
    while(ch>'9' || ch<'0')ch=getchar();
    while(ch>='0' && ch<='9')str=(str<<1)+(str<<3)+ch-48,ch=getchar();
    return str;
}
void link(int x,int y,int z){nxt[++num]=head[x];to[num]=y;dis[num]=z;head[x]=num;}
bool comp(int i,int j){return edg[i]<edg[j];}
int ids=0,fa[N],top[N],id[N],son[N],sz[N];ll path[N];
void dfs1(int x){
    int u;
    sz[x]=1;
    for(int i=head[x];i;i=nxt[i]){
        u=to[i];if(sz[u])continue;
        path[u]=path[x]+dis[i];fa[u]=x;
        val[u]=dis[i];dfs1(u);sz[x]+=sz[u];
        if(sz[u]>sz[son[x]])son[x]=u;
    }
}
void dfs2(int x,int tp){
    id[x]=++ids;top[x]=tp;
    if(son[x])dfs2(son[x],tp);int u;
    for(int i=head[x];i;i=nxt[i]){
        u=to[i];if(u==fa[x] || u==son[x])continue;
        dfs2(u,u);
    }
}
int root[N],totnode=0;
struct node{
    int lazy,l,r;ll sum;
}tr[N*30];
void upd(int rt,int l,int r){
    tr[rt].sum=tr[tr[rt].l].sum+tr[tr[rt].r].sum+tr[rt].lazy*(w[r]-w[l-1]);
}
void updata(int &rt,int last,int l,int r,int sa,int se){
    rt=++totnode;tr[rt]=tr[last];
    if(sa==l && r==se){
        tr[rt].sum+=w[r]-w[l-1];
        tr[rt].lazy++;
        return ;
    }
    int mid=(l+r)>>1;
    if(se<=mid)updata(tr[rt].l,tr[last].l,l,mid,sa,se);
    else if(sa>mid)updata(tr[rt].r,tr[last].r,mid+1,r,sa,se);
    else{
        updata(tr[rt].l,tr[last].l,l,mid,sa,mid);
        updata(tr[rt].r,tr[last].r,mid+1,r,mid+1,se);
    }
    upd(rt,l,r);
}
void updpath(int x){
    int rt=edg[x];
    while(x){
        updata(root[rt],root[rt],1,n,id[top[x]],id[x]);
        x=fa[top[x]];
    }
}
void solve(){
    sort(a+1,a+n+1,comp);
   dfs1(1);dfs2(1,1);
    for(int i=1;i<=n;i++)w[id[i]]=val[i],sed[edg[i]]+=path[i];
    for(int i=1;i<=n;i++)w[i]+=w[i-1],sed[i]+=sed[i-1];
    for(int j=1,i;j<=n;j++){
        i=a[j];
        if(!root[edg[i]])
            root[edg[i]]=root[edg[i]-1];
        updpath(i);
    }
}
ll query(int rt,int l,int r,int sa,int se,int mr){
    if(l==sa && r==se)return tr[rt].sum+(ll)mr*(w[r]-w[l-1]);
    int mid=(l+r)>>1;ll ret=0;mr+=tr[rt].lazy;
    if(se<=mid)ret=query(tr[rt].l,l,mid,sa,se,mr);
    else if(sa>mid)ret=query(tr[rt].r,mid+1,r,sa,se,mr);
    else{
        ret+=query(tr[rt].l,l,mid,sa,mid,mr);
        ret+=query(tr[rt].r,mid+1,r,mid+1,se,mr);
    }
    return ret;
}
ll qtree(int x,int fr){
    if(fr==0)return 0;
    ll ret=0;int rt=root[fr];
    while(x){
        ret+=query(rt,1,n,id[top[x]],id[x],0);
        x=fa[top[x]];
    }
    return ret;
}
void settle(){
   ll ret=0,ans=0,u,x,y,xa,xb;
    while(Q--){
        u=gi();xa=gi();xb=gi();
        x=Min((xa+ans)%A,(xb+ans)%A);
        y=Max((xa+ans)%A,(xb+ans)%A);
        x=lower_bound(b+1,b+tot+1,x)-b;
        y=upper_bound(b+1,b+tot+1,y)-b-1;
        ret=(ll)(sum[y]-sum[x-1])*path[u];
        ret+=sed[y]-sed[x-1];
        ret-=(ll)(qtree(u,y)-qtree(u,x-1))<<1;
        ans=ret;
        printf("%lld\n",ans);
    }
}
void work()
{
    n=gi();Q=gi();A=gi();
    for(int i=1;i<=n;i++)edg[i]=gi(),b[++bt]=edg[i],a[i]=i;
    int x,y,z;
    for(int i=1;i<n;i++){
        x=gi();y=gi();z=gi();
        link(x,y,z);link(y,x,z);
    }
    sort(b+1,b+bt+1);
    tot=unique(b+1,b+bt+1)-b-1;
    for(int i=1;i<=n;i++)edg[i]=lower_bound(b+1,b+tot+1,edg[i])-b;
    for(int i=1;i<=n;i++)sum[edg[i]]++;
    for(int i=1;i<=tot;i++)sum[i]+=sum[i-1];
    solve();settle();
}

int main()
{
    work();
    return 0;
}

转载于:https://www.cnblogs.com/Yuzao/p/7505827.html