【BZOJ3932】任务查询系统,主席树与差分的建树思想

Time:2016.05.08
Author:xiaoyimi
转载注明出处谢谢

传送门
思路:
说真的我一开始做这个题真不知道怎么建树,只知道应该以时间为根建树,记录优先值并维护前缀和,但想不出怎么用到主席树的前缀建树思想,暴力建树+修改肯定是不科学的……
后来去看题解,发现又是差分。
回想起被运输计划支配的恐惧……(还是自己太弱的缘故)
不过这个差分比较简单,因为它正好利用了第i棵主席树以第i-1棵树为基础建树的思路,比如某一任务的时间区间为[l,r],优先度为p,那么我们在root[l]上建树时+p,root[r+1]上建树时-p,这样的话就可以在枚举每个时间点的操作来建树了,因为有m个任务,每个任务有两个时间点的操作,所以最多建2m棵树,每个时间点所代表的主席树就是它最后一次修改所建的树,记录每个时间点最后被操作时所建的树的编号pos[i],查询操作就是空树与pos[i]相减得到了,这是主席树的基本操作,不过多赘述
注意:
1.简单计算可以得到这里的sum最多可以是100000*100000,超过了int,所以sum用long long计算
2.对于每个时间点的修改,有+有-,所以在原先的建树代码上稍作修改即可,记录系数什么的
代码:

#include<bits/stdc++.h>
#define M 100004
#define LL long long
#define ls(x) a[x].ch[0]
#define rs(x) a[x].ch[1] 
using namespace std;
int n,m,cnt,k;
int ID[M],S[M],E[M],P[M],pos[M];
LL lastans=1; 
vector <int> t[M];
struct disc
{
    int data,id;
    bool operator <(const disc other)const
    {
        return data<other.data;
    }
}b[M];
struct Chairman_tree
{
    int siz,ch[2];
    LL sum;
}a[M<<6];
int in()
{
    char ch=getchar();
    int t=0;
    while (!isdigit(ch)) ch=getchar();
    while (isdigit(ch)) t=(t<<3)+(t<<1)+ch-48,ch=getchar();
    return t;
}
void build(int now,int L,int R,int rt,int k,int val)
{
    a[rt].siz=a[now].siz+k;
    a[rt].sum=a[now].sum+k*ID[val];
    if (L==R) return;
    int mid=(L+R)>>1;
    if (mid>=val)
        rs(rt)=rs(now),
        ls(rt)=++cnt,
        build(ls(now),L,mid,ls(rt),k,val);
    else
        ls(rt)=ls(now),
        rs(rt)=++cnt,
        build(rs(now),mid+1,R,rs(rt),k,val);
}
LL get(int begin,int end,int L,int R,int k)
{
    if (begin==end) return a[R].sum-a[L].sum;
    int mid=(begin+end)>>1,t=a[ls(R)].siz-a[ls(L)].siz;
    if (k<=t)
        return get(begin,mid,ls(L),ls(R),k);
    else
        return a[ls(R)].sum-a[ls(L)].sum+get(mid+1,end,rs(L),rs(R),k-t);
}
main()
{
    m=in();n=in();
    for (int i=1;i<=m;i++)
        S[i]=in(),E[i]=in(),P[i]=in(),
        b[i].id=i,b[i].data=P[i];
    sort(b+1,b+m+1);
    for (int i=1;i<=m;i++)
        ID[i]=b[i].data,
        P[b[i].id]=i;
    cnt=1; 
    for (int i=1;i<=m;i++)
        t[S[i]].push_back(P[i]),
        t[E[i]+1].push_back(-P[i]), 
        cnt+=1+(E[i]+1<=n);
    for (int rt=1,i=1;i<=n;i++)
    {
        for (int j=0;j<t[i].size();j++)
            build(rt,1,n,rt+1,t[i][j]/abs(t[i][j]),abs(t[i][j])),
            rt++;
        pos[i]=rt;
    }
    int x;
    for (int i=1;i<=n;i++)
        x=in(),
        k=(lastans*in()+in())%in()+1,
        printf("%lld\n",lastans=get(1,n,1,pos[x],k));
}

你可能感兴趣的:(【BZOJ3932】任务查询系统,主席树与差分的建树思想)