Submission #2208878

---

Source Code Expand

#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
typedef pair<int,int> P;
const ll MOD=10000;
const ll INF=1000000000;
const int MAX=100001;
const double EPS=1e-10;
using namespace std;
struct BIT{
    vector<int>bit;
    int n=1;
    void init(int m){
        while(n<m){
            n*=2;
        }
        bit.resize(n+1);
    }
    int sum(int i){
        int s=0;
        while(i>0){
            s+=bit[i];
            i-=i&-i;
        }
        return s;
    }
    void add(int i,int x){
        while(i<=n){
            bit[i]+=x;
            i+=i&-i;
        }
    }
};
ll b[200010];
map<ll,int> m;
void compress(int n){
    vector<int> v;
    for(int i=0;i<=n;i++){
        v.push_back(b[i]);
    }
    sort(v.begin(),v.end());
    for(int i=0;i<(int)v.size();i++){
        m[v[i]]=i+1;
    }
}
int main() {
    int n;ll k;ll a[200010];
    cin>>n>>k;
    ll sum=0;
    b[0]=0;
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        sum+=a[i];
        b[i]=sum-i*k;
    }
    compress(n);
    BIT bit;
    bit.init(n+1);
    int ans=0;
    for(int i=0;i<=n;i++){
        int x=m[b[i]];
        ans+=bit.sum(x);
        bit.add(x,1);
    }
    cout<<ans<<endl;
    
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:52:28: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld",&a[i]);
                            ^

Judge Result