Cod sursa(job #2587078)

Utilizator AlexandruGabrielAliciuc Alexandru AlexandruGabriel Data 21 martie 2020 23:37:02
Problema Oite Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <bits/stdc++.h>

using namespace std;

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

const int N = 1030, mod = 666013;
int n, a[N], ans, m;

struct oaie
{
    int x;
    int cnt;
};
vector <oaie> H[mod + 1];

void Insert(int val)
{
    int key = val % mod;

    for (int ind = 0; ind < H[key].size(); ind++)
    {
        if (val == H[key][ind].x)
        {
            H[key][ind].cnt++;
            return;
        }
    }

    H[key].push_back({val, 1});
}

void findHash(int val)
{
    int key = val % mod;

    for (int ind = 0; ind < H[key].size(); ind++)
    {
        if (val == H[key][ind].x)
            ans += H[key][ind].cnt;
    }
}

int main()
{
    fin >> n >> m;
    for (int i = 1; i <= n; i++)
        fin >> a[i];

    for (int i = 1; i < n; i++)
    {
        for (int j = i + 1; j <= n; j++)
             if (a[i] + a[j] <= m)
                    findHash(m - a[i] - a[j]);

        for (int j = 1; j < i; j++)
            if (a[i] + a[j] <= m)
                Insert(a[i] + a[j]);
    }

    fout << ans;
    return 0;
}