Cod sursa(job #2035309)

Utilizator Dragos123Tatar Dragos Vlad Dragos123 Data 9 octombrie 2017 10:39:52
Problema Numarare triunghiuri Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <iostream>
#include <algorithm>
using namespace std;

const int Dim = 801;
int a[Dim];
int n;

void ReadArray();
void Rezolvare();
int BinarySearch(int l, int r, int vl, int vr);

int main()
{
    ReadArray();
    Rezolvare();
}

void ReadArray()
{
    cin >> n;
    for(int i = 1; i <= n; ++i)
        cin >> a[i];
}

void Rezolvare()
{
    sort (a + 1, a + n + 1);
    int nr = 0;
    for(int i = 1; i <= n - 2; ++i)
        for(int j = i + 1; j <= n - 1; ++j)
        {
            if (BinarySearch(j + 1, n, a[i], a[j]) != 0)
                nr += BinarySearch(j + 1, n, a[i], a[j]) - j;
        }
    cout << nr;
}

int BinarySearch(int l, int r, int vl, int vr)
{
    int mid;
    int s = vl + vr;
    while (l <= r)
    {
        mid = ( l + r ) / 2;
        if (a[mid] <= s)
            return mid;
        if (a[mid] > s)
            r = mid - 1;
    }
    return 0;
}