Submission #605763

#	Time	Username	Problem	Language	Result	Execution time	Memory
605763		ukraine	Count Squares (CEOI19_countsquares)	C++17	100 / 100	1709 ms	211288 KiB

countsquares

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
#define fi first
#define se second
using namespace std;

ll h, v, x[10005], y[10005], ans = 0;
map <ll, ll> mpx, mpy;

int main() {
    ios_base::sync_with_stdio(NULL); cin.tie(nullptr); cout.tie(nullptr);
    cin >> h >> v;
    for(int i = 1; i <= h; i++) {
        cin >> y[i];
    }
    for(int i = 1; i <= v; i++) {
        cin >> x[i];
    }
    for(int i = 1; i < h; i++) {
        for(int j = i + 1; j <= h; j++) {
            mpy[y[j] - y[i]]++;
        }
    }
    for(int i = 1; i < v; i++) {
        for(int j = i + 1; j <= v; j++) {
            mpx[x[j] - x[i]]++;
        }
    }
    for(auto u: mpx) {
        if(u.se) {
            ans += u.se * mpy[u.fi];
        }
    }
    cout << ans;
}

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