Submission #73240229

---

Source Code Expand

#include <bits/stdc++.h>
#define int long long
using namespace std;
namespace IO {
	template<typename T> inline void read(T &x) {
		bool f = 1;
		x = 0;
		char ch = getchar();
		while (ch < '0' || ch > '9') {
			if (ch == '-')f = 0;
			ch = getchar();
		}
		while (ch >= '0' && ch <= '9') x = (x << 1) + (x << 3) + (ch & 15), ch = getchar();
		x = f ? x : -x;
	}
	template<typename T> inline void write(T x) {
		if (x < 0) putchar('-'), x = -x;
		if (x > 9) write(x / 10);
		putchar(('0' + x % 10));
	}
	template <typename T, typename ...Args> inline
	void read(T &x, Args &...args) {
		read(x), read(args...);
	}
	template<typename T> inline void write(T x, char c) {
		write(x), putchar(c);
	}
}
using namespace IO;
const int INF = 1e18;

int N, K;

signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);

	cin >> N >> K;
	unordered_map<int, int> cnt;
	cnt.reserve(1 << 18);
	cnt.max_load_factor(0.7);
	int x = 0;
	cnt[0] = 1;
	int ans = 0;
	for (int i = 0; i < N; ++i) {
		int a;
		cin >> a;
		x += a;
		auto it = cnt.find(x - K);
		if (it != cnt.end()) ans += it->second;
		cnt[x] += 1;
	}
	cout << ans << '\n';

	return 0;
}

Submission Info

Judge Result