維護一個可以插入刪除的有序序列，每次詢問序列中位置mod5=3的數的和。

　　CodeForces原題，因為時限給的太寬，數據太水，STL可以暴力過。

　　用線段樹和平衡樹都可以做這題，線段樹需要先離散化，然后每個區間五個標記分別保存到這個區間左端點的距離mod5=0~4的和，以及一個標記保存這個區間數的個數。合并區間的時候，左兒子因為左端點相同可以直接合并，而右兒子需要推一下，假設左兒子有Num個元素，右兒子本來Mod5=K的點,合并之后到左端點Mod5的值為(num%5+K)%5，所以sum[p][i] = sum[p<<1][i] + sum[p<<1|1][(i-num[p<<1]%5+5) % 5]。

　　換了下代碼風格，該加空格的地方加上了空格，盡量讓自己的代碼寫的標準一些吧。

1 #include <string.h> 2 #include <stdio.h> 3 #include <algorithm> 4 #define calm l + r >> 1; 5 #define lson l, m, p << 1 6 #define rson m+1, r, p << 1 | 1 7 #define MAXN 100005 8 typedef long long LL; 9 int n,tot,d[MAXN],unq[MAXN],num[MAXN<<2]; 10 char s[MAXN][10]; 11 LL sum[MAXN<<2][5]; 12 void pushup(int p) { 13 for (int i = 0; i < 5; i++) 14 sum[p][i] = sum[p<<1][i] + sum[p<<1|1][(i-num[p<<1]%5+5) % 5]; 15 } 16 void update(int x, int v, int o,int l, int r, int p) { 17 num[p] += o ? 1 : -1; 18 if (l == r) { 19 sum[p][0] = o ? v : 0; 20 return; 21 } 22 int m=calm; 23 if(x <= m)update(x, v, o, lson); 24 else update(x, v, o, rson); 25 pushup(p); 26 } 27 int main() { 28 //freopen ("test.in" ,"r", stdin); 29 while (scanf("%d", &n) != EOF) { 30 tot = 0; 31 for (int i = 0; i < n; i++) { 32 scanf("%s", s[i]); 33 if (*s[i] != 's') 34 scanf("%d", &d[i]), unq[tot++] = d[i]; 35 } 36 std::sort(unq, unq + tot); 37 tot = std::unique(unq, unq + tot) - unq; 38 memset(sum, 0, sizeof sum); 39 memset(num, 0, sizeof num); 40 for (int i = 0; i < n; i++) { 41 if (*s[i] == 's') printf("%I64d\n", sum[1][2]); 42 else { 43 int x = std::lower_bound(unq, unq + tot, d[i]) - unq; 44 update(x, d[i], *s[i] == 'a' ? 1: 0, 1, tot ,1); 45 } 46 } 47 } 48 return 0; 49 }

轉載于:https://www.cnblogs.com/swm8023/archive/2012/09/20/2694591.html

總結

以上是生活随笔為你收集整理的HDU 4288 Coder [线段树]的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。