HNOI2015 开店

Description

 

Input


Output

 

Sample Input

10 10 10
0 0 7 2 1 4 7 7 7 9 
1 2 270
2 3 217
1 4 326
2 5 361
4 6 116
3 7 38
1 8 800
6 9 210
7 10 278
8 9 8
2 8 0
9 3 1
8 0 8
4 2 7
9 7 3
4 7 0
2 2 7
3 2 1
2 3 4

Sample Output

1603
957
7161
9466
3232
5223
1879
1669
1282
0
 

Data Constraint

 

这道题和ZJOI那题点剖比较像。

可以用相似的方法做,先建出一棵重心树。每个点维护一棵动态开节点的线段树,维护重心管辖区域内的点到重心/(重心树上父亲的距离之和),查询的时候沿着重心树跳即可,复杂度O(Nlog2N)。

但是直接做是会爆空间的!!!!

我们需要压缩空间。首先年龄可以离散化,

然后由于每个点最多只有3个儿子,所以重心管辖区域内的点到重心/(重心树上父亲的距离之和)只记录一个即可,成功将空间压至460MB.

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<map>

using namespace std;
typedef long long ll;

map<int,int> H;

struct Tree{
    int size,ls,rs;
    ll sf;
}tr[22000011];

int ds[19][150011];
int G[150011],Next[150011],Y[150011],fs[150011];
int g[150011],num[300011],next[300011],y[300011],len[300011];
int dep[150011],d[150011],fa[18][150011],que[150011];
int size[150011],F[150011],val[150011],root[150011];
int tt,n,m,lim,l,r,a,b,rs,i,x,z,q,j,u,dd,tl,tc,df,tot,yr,T;
bool p[150011];
ll ans;

void st(int i,int j)
{
    T++;
    Next[T]=G[i];
    G[i]=T;
    Y[T]=j;
}

void star(int i,int j,int k)
{
    tt++;
    next[tt]=g[i];
    g[i]=tt;
    y[tt]=j;
    len[tt]=k;
}

int bfs(int st)
{
    tl++;
    int i,ff,l,r,x,j,k,sum,mx;
    l=r=1;
    fs[st]=0;
    que[l]=st;
    while(l<=r){
        x=que[l];
        j=g[x];
        while(j!=0){
            k=y[j];
            if(k!=fs[x]&&p[k]){
                r++;
                que[r]=k;
                fs[k]=x;
                if(tl==1){
                    fa[0][k]=x;
                    dep[k]=dep[x]+len[j];
                    d[k]=d[x]+1;
                }
            }
            j=next[j];
        }
        l++;
    }
    for(i=1;i<=r;i++)size[que[i]]=0;    
    for(i=r;i>=2;i--){
        x=que[i];ff=fs[x];
        size[x]++;
        size[ff]+=size[x];
    }
    size[que[1]]++;    
    sum=size[que[1]];
    mx=0;
    for(i=1;i<=r;i++){
        x=que[i];
        j=g[x];
        mx=0;
        while(j!=0){
            k=y[j];
            if(fs[k]==x&&p[k]){
                if(size[k]>mx)mx=size[k];
            }
            j=next[j];
        }
        if(sum-size[x]>mx)mx=sum-size[x];
        if(mx<=sum/2)return x;
    }
}

int dfs(int x)
{
    int j,k,ls,vs;
    ls=bfs(x);
    p[ls]=false;
    j=g[ls];
    while(j!=0){
        k=y[j];
        if(p[k]){
            vs=dfs(k);
            F[vs]=ls;
            st(ls,vs);
        }
        j=next[j];
    }
    return ls;
}

int get(int x,int z)
{
    int i,l,e;
    if(d[x]<d[z])swap(x,z);
    l=d[x]-d[z];
    e=0;
    while(l){
        if(l%2==1)x=fa[e][x];
        l/=2;
        e++;
    }
    if(x==z)return x;
    for(i=17;i>=0;i--)if(fa[i][x]!=fa[i][z]){
        x=fa[i][x];
        z=fa[i][z];
    }
    return fa[0][x];
}

int dis(int x,int z)
{
    return dep[x]+dep[z]-2*dep[get(x,z)];
}

void insert(int &t,int l,int r,int x,int nf)
{
    if(t==0)t=++tc;
    if(l==r){
        tr[t].size++;
        tr[t].sf+=nf;
        return;
    }
    int mid;
    mid=(l+r)/2;
    if(x<=mid)insert(tr[t].ls,l,mid,x,nf);
    if(x>mid)insert(tr[t].rs,mid+1,r,x,nf);
    tr[t].sf=tr[tr[t].ls].sf+tr[tr[t].rs].sf;
    tr[t].size=tr[tr[t].ls].size+tr[tr[t].rs].size;
}

Tree Merge(Tree a,Tree b)
{
    Tree c;
    c.size=a.size+b.size;
    c.sf=a.sf+b.sf;
    return c;
}

Tree ask(int t,int l,int r,int x,int y)
{
    if(t==0)return tr[t];
    if(l==x&&r==y)return tr[t];
    int mid;
    mid=(l+r)/2;
    if(y<=mid)return ask(tr[t].ls,l,mid,x,y);
    if(x>mid)return ask(tr[t].rs,mid+1,r,x,y);
    if(x<=mid&&y>mid)return Merge(ask(tr[t].ls,l,mid,x,mid),ask(tr[t].rs,mid+1,r,mid+1,y));
}

void Find(int x,int l,int r)
{
    int dx,rl,nxt,j,k,st;
    nxt=0;
    Tree ts;
    rl=x;
    st=0;
    while(x){
        dx=ds[st][rl];
        j=G[x];
        while(j!=0){
            k=Y[j];
            if(k!=nxt){
                ts=ask(root[k],1,tot,l,r);
                ans+=ts.sf+(ll)ts.size*dx;
            }
            j=Next[j];
        }
        if(val[x]>=l&&val[x]<=r)ans+=dx;
        nxt=x;
        x=F[x];
        st++;
    }
}

int GG(int x)
{
    int l,r,mid;
    l=1;
    r=tot;
    while(l<=r){
        mid=(l+r)/2;
        if(num[mid]>=x)r=mid-1;
        else l=mid+1;
    }
    return l;
}

int main()
{
    memset(p,true,sizeof(p));
    scanf("%d%d%d",&n,&m,&lim);
    for(i=1;i<=n;i++){
        scanf("%d",&val[i]);
        val[i]++;
        if(!H[val[i]]){
            H[val[i]]=1;
            num[++tot]=val[i];
        }
    }
    sort(num+1,num+1+tot);
    for(i=1;i<=tot;i++)H[num[i]]=i;
    for(i=1;i<=n;i++)val[i]=H[val[i]];
    for(i=1;i<n;i++){
        scanf("%d%d%d",&x,&z,&q);
        star(x,z,q);
        star(z,x,q);
    }
    rs=dfs(1);
    for(i=1;i<=17;i++)
        for(j=1;j<=n;j++)fa[i][j]=fa[i-1][fa[i-1][j]];
    for(i=1;i<=n;i++){
        x=i;
        l=0;
        while(x){
            ds[l][i]=dis(i,x);
            x=F[x];
            l++;
        }
    }
    for(i=1;i<=n;i++){
        l=0;
        x=i;
        while(x){
            if(F[x])df=ds[l+1][i];
            else df=0;
            insert(root[x],1,tot,val[i],df);
            x=F[x];
            l++;
        }
    }
    ans=0;
    for(i=1;i<=m;i++){
        scanf("%d%d%d",&u,&a,&b);
        a=(a+ans)%lim;b=(b+ans)%lim;
        a++;b++;
        l=min(a,b);r=max(a,b);yr=r;
        l=GG(l);r=GG(r);
        if(r>tot||num[r]>yr)r--;
        ans=0;
        if(l<=r)Find(u,l,r);
        printf("%lld\n",ans);
    }
}

 

posted on 2015-04-27 22:13  razorjxt  阅读(213)  评论(0编辑  收藏  举报