Submission #605757

#	Time	Username	Problem	Language	Result	Execution time	Memory
605757		ukraine	Count Squares (CEOI19_countsquares)	C++17	7 / 100	177 ms	31256 KiB

countsquares

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

ll h, v, x[10005], y[10005], ans = 0;
set <ll> sy;
map <ll, ll> mp;

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++) {
            sy.insert(y[j] - y[i]);
        }
    }
    for(int i = 1; i < v; i++) {
        for(int j = i + 1; j <= v; j++) {
            mp[x[j] - x[i]]++;
        }
    }
    for(auto u: sy) {
        if(mp[u]) ans += mp[u];
    }
    cout << ans;
}

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