问题 D: 清点人数
时间限制: 1 Sec 内存限制: 128 MB
提交: 94 解决: 16
[提交][状态][讨论版][命题人:quanxing][Edit] [TestData]
题目链接:http://acm.ocrosoft.com/problem.php?cid=1688&pid=3
题目描述
初始时,火车上没有学生;当同学们开始上火车时,年级主任从第一节车厢出发走到最后一节车厢,每节车厢随时都有可能有同学上下。年级主任走到第m节车厢时,他想知道第1到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就输出多少行,每行一个整数,回答年级主任提出的问题。
样例输入
10 7
A 1
B 1 1
B 3 1
B 4 1
A 2
A 3
A 10
样例输出
0
1
2
3
思路:数组数组模板题,输入用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);
}
}
}