问题 D: 清点人数

时间限制: 1 Sec 内存限制: 128 MB
提交: 94 解决: 16
[提交][状态][讨论版][命题人:quanxing][Edit] [TestData]

题目链接：http://acm.ocrosoft.com/problem.php?cid=1688&pid=3

题目描述

初始时，火车上没有学生；当同学们开始上火车时，年级主任从第一节车厢出发走到最后一节车厢，每节车厢随时都有可能有同学上下。年级主任走到第m节车厢时，他想知道第１到m这m节车厢上一共有多少学生，但是他没有调头往回走的习惯．也就是说每次当他提问时，m总会比前一次大。

输入

第一行两个整数n,k，表示火车共有n节车厢以及k个事件。接下来有k行，按时间先后给出k个事件，每行开头都有一个字母A,B或C，如果字母为A，接下来是一个数m，表示年级主任现在在第m节车厢；如果为B，接下来两个数m,p，表示在第m节车厢有p名学生上车；如果为C，接下来两个数m,p，表示在第m节车厢有p名学生下车。学生总人数不会超过100000。

注意:对于30%的数据,n<=10000,k<=10000 至少有3000个A
对于100%的数据n<=500000,k<=100000. 至少有30000个A

输出

有多少个A就输出多少行，每行一个整数，回答年级主任提出的问题。

样例输入

样例输出

思路：数组数组模板题，输入用scanf(“%s”,&ch),不然会卡时间，但我也不知道为什么

代码：

#include<bits/stdc++.h>

using namespace std;

#define maxn 1000000

int A[maxn], C[maxn];

int n, m;

int lowbit(int t)//t转换为二进制后，有末尾连续0共k个，返回2^k次方

{

    return t & -t;

}

void add(int x, int y)//单点修改

{

    for (int i = x; i <= n; i += lowbit(i))

    {

         C[i] += y;

    }

}

int sum(int x)//区间求和

{

    int ans = 0;

    for (int i = x; i > 0; i -= lowbit(i))

    {

         ans += C[i];

    }

    return ans;

}

int main()

{

    char ch[2];

    int a, b, mm;

    scanf("%d %d", &n, &m);

    //getchar();

    while (m--)

    {



         scanf("%s", &ch);

         //getchar();

         if (ch[0] == 'A')

         {

             scanf("%d", &mm);

             printf("%d\n", sum(mm));

         }

         else if (ch[0] == 'B')

         {

             scanf("%d %d", &a, &b);

             add(a, b);

         }

         else

         {

             scanf("%d %d", &a, &b);

             add(a, -b);

         }

    }

}