CSU 1724 相等距离的和(离线+线段树)

题意:给你一些操作,给集合里加数字,给集合里删除数字,求和是对从x开始的,间隔L的数字的和,数字放进集合内就被自动排为升序,对于每次求和输出结果

题解:因为数字很大所以需要离散化,先离线离散化,然后就是直接更新,但是因为要间隔求和,并且集合里数字的个数会变化,所以需要在每个节点维护L个值,这L个值是在这个节点(包含一些升序的数),从x位开始的节点内的和,怎么理解呢,如果这个节点有数字1 2 3 4 5 ,间隔为2,sum[rt][0]=1+3+5,sum[rt][1]=2+4,这样的意思

所以每次更新的时候,就是直接把更新的值修改在单点sum[rt][0]上即可,然后就是一段节点的sum怎么更新,sum[rt][i],显然在左儿子里面,需要取sum[lrt][i],为什么呢,因为rt这个节点和lrt这个节点的开头是一起的,所以rt节点从i开始,等于lrt节点从i开始,然后就是rrt节点里应该从哪一位开始加,lrt里有一些数字,然后取了i,i+L。。。,看最后有几个没被取,然后L-这个值就是rrt里面的开头,怎么求这个开头呢,要记录lrt节点里有多少个值num[lrt],然后假设从求i位的和,((num[lrt]-(i+1))%L+L)%L,这个就是说明lrt节点最后有几个没被选,然后L-((num[lrt]-(i+1))%L+L)%L -1就是在rrt里面的开头位置(位置从0到L-1),这样就OK了啊;

可能说的有点抽象,所以需要画图好好分析。

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           100005
#define   MAXN          500005
#define   maxnode       105
#define   sigma_size    2
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-9;
const LL     mod   = 1e9+7;
const ull    mx    = 1e9+7;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
}
/*****************************************************/

char s[MAX][10];
int a[MAX];
int b[MAX];
LL sum[MAX<<2][5];
int numb[MAX<<2];
int n,L;

void pushup(int rt){
    numb[rt]=numb[lrt]+numb[rrt];
    for(int i=0;i<L;i++){
        sum[rt][i]=sum[lrt][i]+sum[rrt][L-((numb[lrt]-(i+1))%L+L)%L-1];
    }
}
void build(int l,int r,int rt){
    numb[rt]=0;
    for(int i=0;i<L;i++) sum[rt][i]=0;
    if(l==r) return;
    middle;
    build(lson);
    build(rson);
}

void update(int l,int r,int rt,int pos,int d){
    if(l==r){
        sum[rt][0]+=d;
        if(sum[rt][0]) numb[rt]++;
        else numb[rt]--;
        return ;
    }
    middle;
    if(pos<=m) update(lson,pos,d);
    else update(rson,pos,d);
    pushup(rt);
}

int main(){
    //freopen("test.txt","r",stdin);
    int kase=0;
    while(cin>>n>>L){
        int num=0;
        kase++;
        for(int i=0;i<n;i++){
            scanf("%s%d",s[i],&a[i]);
            if(s[i][0]!='s') b[num++]=a[i];
        }
        sort(b,b+num);
        int tot=unique(b,b+num)-b;
        build(1,n,1);
        printf("Case %d:\n",kase);
        for(int i=0;i<n;i++){
            if(s[i][0]=='s') printf("%lld\n",sum[1][a[i]-1]);
            else{
                int pos=lower_bound(b,b+num,a[i])-b;
                if(s[i][0]=='d') a[i]=-a[i];
                update(1,n,1,pos+1,a[i]);
            }
        }
    }
    return 0;
}










你可能感兴趣的:(ACM)