Cod sursa(job #2429949)

Utilizator IulianOleniucIulian Oleniuc IulianOleniuc Data 11 iunie 2019 22:35:57
Problema Oite Scor 70
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <bits/stdc++.h>
using namespace std;

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

class HashTable {
  private:
    int MOD;
    vector<vector<pair<int, int>>> table;

  public:
    HashTable(int MOD) {
        this->MOD = MOD;
        table.resize(MOD);
    }

    int& operator[](int val) {
        int hash = val % MOD;
        for (auto& it : table[hash])
            if (it.first == val)
                return it.second;

        table[hash].push_back(make_pair(val, 0));
        return table[hash].back().second;
    }
};

int main() {
    int n, s;
    fin >> n >> s;

    vector<int> v(n + 1);
    for (int i = 1; i <= n; i++)
        fin >> v[i];

    int64_t sol = 0;
    HashTable sums(666013);

    sums[v[1] + v[2]]++;
    for (int i = 3; i < n; i++) {
        for (int j = i + 1; j <= n; j++)
            if (s >= v[i] + v[j])
                sol += sums[s - v[i] - v[j]];
        for (int j = i - 1; j >= 1; j--)
            sums[v[i] + v[j]]++;
    }

    fout << sol << '\n';
    fout.close();
    return 0;
}