这题我用了两种方法,都超时,最后百度一个出来.....,上传两个代码,第一个代码是网上查的,那三个函数表示有点不懂,求解答!
描述
初始时,火车上没有学生;当同学们开始上火车时,年级主任从第一节车厢出发走到最后一节车厢,每节车厢随时都有可能有同学上下。年级主任走到第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。
输出
有多少个A就输出多少行,每行一个整数,回答年级主任提出的问题。
样例输入
10 7
A 1
B 1 1
B 3 1
B 4 1
A 2
A 3
A 10
A 1
B 1 1
B 3 1
B 4 1
A 2
A 3
A 10
样例输出
0
1
2
3
1
2
3
提示
注意:对于30%的数据,n< =10000,k< =10000 至少有3000个A 对于100%的数据n< =500000,k< =100000. 至少有30000个A
#include<iostream> #include<string.h> #include<stdio.h> #include<stdlib.h> long c[500001]; long n,k; long fun(long x) { return x^(x&(x-1)); } void modify(long x,long d) { while (x<=n) { c[x]+=d; x+=fun(x); } } long getsum(long x) { long s=0; while (x) { s+=c[x]; x-=fun(x); } return s; } int main() { scanf("%d%d",&n,&k); getchar(); memset(c,0,sizeof(c)); long x,y; char c; for (int i=1;i<=k;i++) { scanf("%c",&c); if (c=='A') { scanf("%d",&x); getchar(); printf("%d\n",getsum(x)); } else if (c=='B') { scanf("%d%d",&x,&y); getchar(); modify(x,y); } else { scanf("%d%d",&x,&y); getchar(); modify(x,-y); } } return 0; }
/*#include<iostream> #include<cstring> #include<cstdio> using namespace std; int a[1000000]; int main() { int n,k; char c; while(scanf("%d %d",&n,&k)!=EOF) { int sum=0; memset(a,0,sizeof(a)); for(int i=0;i<k;i++) { cin>>c; if(c=='A') { int ms; scanf("%d",&ms); printf("%d\n",a[ms]); } if(c=='B') { int ms,p; scanf("%d%d",&ms,&p); for(int ii=ms;ii<=n;ii++) a[ii]+=p; } if(c=='C') { int ms,p; scanf("%d%d",&ms,&p); for(int ii=ms;ii<=n;ii++) a[ii]-=p; } } } return 0; } */