Cod sursa(job #2930410)

Utilizator Gica-gicutaGeorge Gica-gicuta Data 28 octombrie 2022 13:12:05
Problema Oite Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.75 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("oite.in");
ofstream fout("oite.out");
int v[1025];
unordered_map <int,int> f;
int main()
{
    ios_base::sync_with_stdio(false);
    fin.tie();
    fout.tie();
    int n,l;
    long long cnt=0;
    fin>>n>>l;
    for(int i=1; i<=n; i++)
        fin>>v[i];
    sort(v+1,v+n+1);
    for(int i=2; i<=n-2; i++)
    {
        for(int j=1; j<=i-1; j++)
        {
            f[v[i]+v[j]]++;
            if(v[i]+v[j]>l)break;
        }
        for(int j=i+2; j<=n; j++)
        {
            if(f.find(l-v[i+1]-v[j])!=f.end())
            {
                cnt+=f[l-v[i+1]-v[j]];
            }
            if(v[i+1]+v[j]>l)break;
        }
    }
    fout<<cnt;
    return 0;
}