bzoj3932: [CQOI2015]任務查詢系統(主席樹)

Hanks_o發表於2018-04-11

題目傳送門

解法:
求一個點的狀態。
這個就差分下然後用主席樹維護就可以了啊。

l到r加上k這個任務。
那麼l加上k的影響,r+1減去k的影響。
然後字首和就是每個點的狀態。
那麼直接差分主席樹。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
typedef long long ll;
struct node {int lc,rc,c;ll s;}t[5100000];int cnt,rt[110000];
void build(int &u,int l,int r,int p,int c,ll s) {
    if(u==0)u=++cnt;t[u].c+=c;t[u].s+=s;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(t[u].lc,l,mid,p,c,s);
    else build(t[u].rc,mid+1,r,p,c,s);
}
void Merge(int &u1,int u2) {
    if(u1==0) {u1=u2;return ;}if(u2==0)return ;
    t[u1].c+=t[u2].c;t[u1].s+=t[u2].s;
    Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
ll find(int u,int l,int r,int k) {
    if(k>t[u].c)return t[u].s;
    if(l==r)return ll(k*(t[u].s/t[u].c));int mid=(l+r)/2;
    int lc=t[u].lc,rc=t[u].rc;
    if(t[lc].c>=k) return find(t[u].lc,l,mid,k);
    else return find(t[u].rc,mid+1,r,k-t[lc].c)+t[lc].s;
}
ll s[110000],sum[110000];
struct edge {int a,b;ll c;}a[110000];
bool cmp(edge n1,edge n2) {return n1.c<n2.c;}
int main() {
    int n,m;scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++) scanf("%d%d%lld",&a[i].a,&a[i].b,&a[i].c);
    sort(a+1,a+1+m,cmp);int tot=0;
    for(int i=1;i<=m;i++) {
        if(a[i].c!=a[i-1].c)tot++;
        build(rt[a[i].a],1,n,tot,1,a[i].c);build(rt[a[i].b+1],1,n,tot,-1,-a[i].c);
    }
    for(int i=1;i<=n;i++)Merge(rt[i],rt[i-1]);
    ll pre=1;
    for(int i=1;i<=n;i++) {
        int x,a,b,c;scanf("%d%d%d%d",&x,&a,&b,&c);
        pre%=c;a%=c;b%=c;int k=1+(a*pre+b)%c;
        pre=find(rt[x],1,n,k);printf("%lld\n",pre);
    }
    return 0;
}

相關文章