Coder

Coder

Time Limit: 20000/10000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 7069    Accepted Submission(s): 2635

Problem Description

　　In mathematics and computer science, an algorithm describes a set of procedures or instructions that define a procedure. The term has become increasing popular since the advent of cheap and reliable computers. Many companies now employ a single coder to write an algorithm that will replace many other employees. An added benefit to the employer is that the coder will also become redundant once their work is done. ¹
　　You are now the signle coder, and have been assigned a new task writing code, since your boss would like to replace many other employees (and you when you become redundant once your task is complete).
Your code should be able to complete a task to replace these employees who do nothing all day but eating: make the digest sum.
　　By saying “digest sum” we study some properties of data. For the sake of simplicity, our data is a set of integers. Your code should give response to following operations:
　　1. add x – add the element x to the set;
　　2. del x – remove the element x from the set;
　　3. sum – find the digest sum of the set. The digest sum should be understood by

　　where the set S is written as {a₁, a₂, ... , a_k} satisfying a₁ < a₂ < a₃ < ... < a_k 
　　Can you complete this task (and be then fired)?
------------------------------------------------------------------------------
¹ See http://uncyclopedia.wikia.com/wiki/Algorithm

 

Input

　　There’re several test cases.
　　In each test case, the first line contains one integer N ( 1 <= N <= 10⁵ ), the number of operations to process.
　　Then following is n lines, each one containing one of three operations: “add x” or “del x” or “sum”.
　　You may assume that 1 <= x <= 10⁹.
　　Please see the sample for detailed format.
　　For any “add x” it is guaranteed that x is not currently in the set just before this operation.
　　For any “del x” it is guaranteed that x must currently be in the set just before this operation.
　　Please process until EOF (End Of File).

 

Output

　　For each operation “sum” please print one line containing exactly one integer denoting the digest sum of the current set. Print 0 if the set is empty.

 

Sample Input

9 add 1 add 2 add 3 add 4 add 5 sum add 6 del 3 sum 6 add 1 add 3 add 5 add 7 add 9 sum

 

Sample Output

3 4 5

Hint

C++ maybe run faster than G++ in this problem.

 题意：

有三种类型的操作，（1）."add x"，表示往集合里添加数x。（2）.“del x”表示将集合中数x删除。（3）.“sum”求出从小到大排列的集合中下标模5为3的数的和。集合中的数都是唯一的

 

这里由于要进行插入和删除操作，肯定无法在线段树上进行在线操作，所以我们采用离线的方法。

先把所有的数据存在a数组里面，并把a数组的东西复制b数组里，对b数组进行离散化，根据b数组开始建树，在线段树维护两个值，num和s，num表示区间里元素的个数，s[i]代表区间里

所有满足下标x％5=i的元素和。

所以每次更新的时候，都是

                               t[rt].s[i] = t[rt<<1].s[i] + t[rt<<1|1].s[((i - t[rt<<1].num) % 5 + 5) % 5];

 

因为右结点的第x元素下标在父结点中应该对应的是第(x+左结点的元素个数)个元素(i为右边结点的下标)，可以自己模拟下，就会了，即i=(x+cnt)%5,x=(5*n+i-cnt)%5，取模是为了防止负数

每次查询时只需要返回node[1].sum[2]即可

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define ll long long
const int maxn = 1e5+7;
char str[maxn][4];
int a[maxn],b[maxn];
struct node
{
    int l, r;
    ll s[10];
    int num;
} t[maxn * 4];
void build(int rt, int l, int r)
{
    t[rt].l = l;
    t[rt].r = r;
    memset(t[rt].s, 0, sizeof(t[rt].s));
    if(l == r)
    {
        t[rt].num = 0;
        return ;
    }
    int mid = (l + r) >> 1;
    build(rt<<1, l, mid);
    build(rt<<1|1, mid + 1, r);
    t[rt].num = 0;
   
    memset(t[rt].s, 0, sizeof(t[rt].s));
    
}
void update(int rt, int loc, int val, int op)
{
    op == 1 ? t[rt].num ++ : t[rt].num--;
    if(t[rt].l == t[rt].r)
    {
        t[rt].s[0] = op * val;
        return ;
    }
    int mid = (t[rt].l + t[rt].r) >> 1;
    if(loc <= mid)
        update(rt<<1, loc, val, op);
    else
        update(rt<<1|1, loc, val, op);
    for(int i = 0; i < 5; i++)
    {
        t[rt].s[i] = t[rt<<1].s[i] + t[rt<<1|1].s[((i - t[rt<<1].num) % 5 + 5) % 5];
    }
}

int main()
{

    int n;
    while(~scanf("%d", &n))
    {
        int len = 0;
       
        for(int i = 0; i < n; i ++)
        {
            scanf("%s", str[i]);
            if(str[i][0] != 's')
            {
                scanf("%d", &a[i]);
                b[len ++] = a[i];
            }
        }
        sort(b,b+len);
        len = unique(b,b+len) - b;
        build(1,1,len);
        for(int i = 0; i < n; i ++)
        {
            if(str[i][0] == 'a')
            {
                int x=lower_bound(b,b+len,a[i])-b+1;
                update(1, x, a[i], 1);
            }
            if(str[i][0] == 'd')
            {
                int x=lower_bound(b,b+len,a[i])-b+1;
                update(1, x, a[i], 0);
            }
            if(str[i][0] == 's')
            {
                printf("%I64d
", t[1].s[2]);
            }
        }


    }
    return 0;
}