1. 程式人生 > >hdu 4288 線段樹+離線處理

hdu 4288 線段樹+離線處理

題意 給你類似一個升序的set結構 add就是加入一個數 del就是刪除一個數 sum就是把所有%5=3的位置的數求和

這題我們怎麼入手呢?以前做過一道類似的樹狀陣列開55個的題 這個題其實大同小異 就是對sum陣列表示%5的值我們開個sum[5]

那麼你既然是升序的 我每次add 左邊的數一定等於sum[i] 主要是右邊的數如何確定 我們知道i = x + cnt(cnt為左子樹的點個數)

那麼x = i - cnt 於是每次up操作我們爸爸的sum等於 左子樹的sum + 右子樹 (i - cnt%5 + 5) %5 

所以up就是這樣

由於數值很大 所以我們要離散化

這是我的程式碼

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <queue>
#include <stack>
#include <queue>
#include <iostream>
#include <sstream>
#include <set>
using namespace std;
#define dbg(x) ;
//cout<<#x<<" = "<< (x)<< endl
const int MAX_N = 1e5+6;
long long sum[MAX_N<<2][5];
long long a[MAX_N],b[MAX_N],num[MAX_N<<2];
int q[MAX_N];
char ch[10];
void up(int rt){
    num[rt] = num[rt<<1]+num[rt<<1|1];
    for(int i = 0;i<5;++i){
        sum[rt][i] = sum[rt<<1][i] + sum[rt<<1|1][((i-num[rt<<1]%5+5)%5)];
    }
}

void update(int rt,int l,int r,int x,int key){
    if(l==r){
        sum[rt][1] = x? 0:b[key];
        num[rt] = !x;
        return ;
    }
    int mid = (l+r)>>1;
    if(key<=mid) update(rt<<1,l,mid,x,key);
    else update(rt<<1|1,mid+1,r,x,key);
    up(rt);
}

int init(int m){
    int top = 0;
    for(int i = 1;i<=m;i++){
        scanf("%s",ch);
        if(ch[0]=='a'){
            q[i] = 0;
            scanf("%lld",&a[i]);
            b[++top] = a[i];
        }
        else if(ch[0]=='d'){
            q[i] = 1;
            scanf("%lld",&a[i]);
        }
        else {
            q[i] = 2;
        }
    }
    sort(b+1,b+1+top);
    memset(num,0,sizeof(num));
    memset(sum,0,sizeof(sum));
    return top;
}

int main(){
    int key,n,m;
    while(~scanf("%d",&m)){
        n = init(m);
        for(int i = 1;i<=m;++i){
            if(q[i]==0){
                int xb = lower_bound(b+1,b+1+n,a[i])-b;
                update(1,1,n,0,xb);
            }
            else if(q[i]==1){
                int xb = lower_bound(b+1,b+1+n,a[i])-b;
                update(1,1,n,1,xb);
            }
            else printf("%lld\n",sum[1][3]);
        }
    }
    return 0;
}