hdu 4288 線段樹+離線處理
阿新 • • 發佈:2018-12-14
題意 給你類似一個升序的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; }