Cod sursa(job #2792312)

Utilizator Stefan_GhinescuGhinescu Stefan-George Stefan_Ghinescu Data 1 noiembrie 2021 13:59:50
Problema Numarare triunghiuri Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <iostream>
#include <algorithm>

using namespace std;

///ifstream fin("nrtri.in");
///ofstream fout("nrtri.out");

const int NMAX = 800;

int v[NMAX + 5];

int bs(int i, int j, int n)
{
    int rez = 0, pas = 1 << 10, x = v[i] + v[j];
    while(pas)
    {
        if (rez + pas <=n && v[rez + pas] <= x)
            rez += pas;
        pas >>= 1;
    }
    return rez;
}

int main()
{
    int n, i, j, k, s = 0;
    cin >> n;
    for (i = 0; i < n; ++i)
    {
        cin >> v[i];
    }
    sort(v, v + n);
    for(i = 0; i < n - 2 ; ++i)
    {
        for(j = i + 1 ; j < n - 1; ++j)
        {
            k = bs(i, j, n);
            if (k > j)
            {
                s += k - j;
            }
        }
    }
    cout << s;
    return 0;
}