Submission #359071

#	Time	Username	Problem	Language	Result	Execution time	Memory
359071		jesus_coconut	Count Squares (CEOI19_countsquares)	C++17	100 / 100	2233 ms	158616 KiB

countsquares

#include <bits/stdc++.h>

using namespace std;

int H, V;
vector<int> h, v;

void read() {
	cin >> H >> V;
	h.resize(H);
	v.resize(V);
	for (auto &a : h) cin >> a;
	for (auto &a : v) cin >> a;
}

map<int, int> calcDiff(vector<int> const &v) {
	map<int, int> ret;
	for (int i = 0; i < (int)v.size(); ++i) {
		for (int j = i + 1; j < (int)v.size(); ++j) {
			ret[v[j] - v[i]]++;
		}
	}
	return ret;
}

void solve() {
	read();
	auto mh = calcDiff(h);
	auto mv = calcDiff(v);
	using ll = long long;
	ll ans = 0;
	for (auto &[key, val] : mh) {
		ans += val * mv[key];
	}
	cout << ans << '\n';
}

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

	solve();

	return 0;
}

• Subtask 1: 7 / 7
• Subtask 2: 40 / 40
• Subtask 3: 53 / 53