Submission #234827

#	Time	Username	Problem	Language	Result	Execution time	Memory
234827		Lyestria	Count Squares (CEOI19_countsquares)	C++14	100 / 100	2041 ms	141212 KiB

countsquares

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int>pii;
typedef long long ll;
typedef pair<ll,ll>pll;
typedef __int128 lll;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const ll mod=1e9+7;
const ll mn=3e5+10;
ll a[mn],b[mn];
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,m;
    cin>>n>>m;
    for(int i=0;i<n;i++)cin>>a[i];
    for(int i=0;i<m;i++)cin>>b[i];
    map<ll,ll>ma;
    for(int i=1;i<n;i++)for(int j=0;j<i;j++)ma[a[i]-a[j]]++;
    ll ans=0;
    for(int i=1;i<m;i++)for(int j=0;j<i;j++)ans+=ma[b[i]-b[j]];
    printf("%lld",ans);
}

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