Submission #3596012

---

Source Code Expand

#include <bits/stdc++.h>
#define st first
#define nd second
#define mp make_pair
#define pb push_back
#define N 1000005
using namespace std;

typedef long long ll;

ll n, k, m, ans, a[N], ne[N], fen[N], pre[N];
map < ll , int > h, hh;
map < ll , int > :: iterator it;

void up(int x, int y){
	for(; x <= m; x += x&-x)
		fen[x] += y;
}

int qu(int x){
	int top = 0;
	for(; x > 0; x -= x&-x)
		top += fen[x];
	return top;
}

int main() {
	// freopen("in.txt", "r", stdin);
	// freopen("amk.txt", "w", stdout);
	scanf("%lld %lld",&n ,&k);
	for(int i = 1; i <= n; i++){
		scanf("%lld",a + i);
		pre[i] = pre[i - 1] + a[i];
	}
	for(int i = 0; i <= n; i++)
		h[i*k - pre[i]]++;
	for(it = h.begin(); it != h.end(); it++){
		ne[++m] = it->st;
		// cout << it->st << " " << m << endl;
		hh[it->st] = m;
	}
	// cout << "amk" << endl;
	up(hh[0], 1);
	for(int i = 1; i <= n; i++){
		int amk = hh[i*k - pre[i]];
		ans += qu(m) - qu(amk - 1);
		up(amk, 1);
	}
	printf("%lld\n", ans);
	return 0;
}

Submission Info

Compile Error

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

Judge Result