我有以下 Fenwick 树算法的实现。
#include <bits/stdc++.h>
#define LSOne(S) (S & (-S))
using namespace std;
typedef long long int L;
int rsq(L b, L x[]){
int sum=0;
for(; b; b-=LSOne(b)) sum+=x[b];
return sum;
}
void adjust(int k, L v, L x[], int n){
for(; k<=n;k+=LSOne(k)) x[k]+=v;
}
int main(){
int n,q; cin>>n>>q;
L ft[n+1]; memset(ft,0,sizeof(ft));
while(q--){
char x; int i,j; cin>>x;
if(x=='+'){
cin>>i>>j;
adjust(i+1,j,ft,n);
} else if(x=='?') {
cin>>i;
cout<<rsq(i,ft)<<endl;
}
}
}
这个程序应该能够运行N<=5000000
和处理Q<=5000000
。它应该能够在 9 秒内运行。但是在提交这个问题后,我得到了超过时间限制(TLE)的判决。我已经尝试了所有措施来优化这段代码,但无济于事,它仍然给了我 TLE。我怎么可能优化这段代码,使它可以在 9 秒内运行。非常感谢。