Cod sursa(job #2846271)

Utilizator cdenisCovei Denis cdenis Data 9 februarie 2022 01:49:17
Problema Oite Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <iostream>
#include <fstream>
#include <unordered_map>

using namespace std;

ifstream fin("oite.in");
ofstream fout("oite.out");

const int MAX=1030;
int n,l,cnt,v[MAX],tmp[MAX];
unordered_map < int, int > ap;

void mergesort(int st, int dr)
{
    if(st==dr)
        return;
    int m=(st+dr)/2;
    mergesort(st,m);
    mergesort(m+1,dr);
    int k=0;
    int i=st;
    int j=m+1;
    while(i<=m && j<=dr)
        if(v[i]<v[j])
            tmp[++k]=v[i++];
        else
            tmp[++k]=v[j++];
    while(i<=m)
        tmp[++k]=v[i++];
    while(j<=dr)
        tmp[++k]=v[j++];
    for(i=st,j=1;i<=dr;i++,j++)
        v[i]=tmp[j];
}

int main()
{
    fin >> n >> l;
    for(int i=1;i<=n;i++)
        fin >> v[i];
    mergesort(1,n);
    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        {
            int s=v[i]+v[j];
            if(ap.find(l-s)!=ap.end())
                cnt+=ap[l-s];
        }
        for(int j=1;j<i;j++)
            ap[v[i]+v[j]]++;
    }
    fout << cnt;
    return 0;
}