維護(hù)一個(gè)可以插入刪除的有序序列，每次詢問(wèn)序列中位置mod5=3的數(shù)的和。

　　CodeForces原題，因?yàn)闀r(shí)限給的太寬，數(shù)據(jù)太水，STL可以暴力過(guò)。

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

　　換了下代碼風(fēng)格，該加空格的地方加上了空格，盡量讓自己的代碼寫(xiě)的標(biāo)準(zhǔn)一些吧。

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 }

轉(zhuǎn)載于:https://www.cnblogs.com/swm8023/archive/2012/09/20/2694591.html

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。