Cod sursa(job #2568239)

Utilizator ancaoxoxSfia Anca ancaoxox Data 3 martie 2020 21:36:42
Problema Oite Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.12 kb
#include <iostream>
#include <vector>
#define MOD 100003

using namespace std;

ifstream cin("oite.in");
ofstream cout("oite.out");

int n, s, l, sol, q;
int v[2005];

vector <pair <int, int> > h[MOD];

int main()
{
    cin >> n >> l;
    for (int i=1; i<=n; i++)
    {
        cin >> v[i];
    }
    for (int i=1; i<n; i++)
    {
        for (int j=i+1; j<=n; j++)
        {
            s = l-v[i]-v[j];
            for (q=0; q<h[s%MOD].size(); q++)
            {
                if (h[s%MOD][q].first == s)
                {
                    sol=sol+h[s%MOD][q].second;
                    break;
                }
            }
        }
        for (int j=1; j<i; j++)
        {
            s=v[i]+v[j];
            for (q=0; q<h[s%MOD].size(); q++)
            {
                if (h[s%MOD][q].first == s)
                {
                    h[s%MOD][q].second++;
                    break;
                }
            }
            if (q == h[s%MOD].size())
            {
                h[s%MOD].push_back({s, 1});
            }
        }
    }
    cout << sol;
    return 0;
}