繁体   English   中英

优化 Fenwick 树 (C++)

[英]Optimizing Fenwick Tree (C++)

我有以下 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 秒内运行。 非常感谢。

从标准输入读取 500000 行所需的时间可能是问题所在。 您是否尝试优化 IO 缓冲:

int main() {
    cin.tie(nullptr);
    std::ios::sync_with_stdio(false);
    ...

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM