Cod sursa(job #2251382)

Utilizator andreistanStan Andrei andreistan Data 1 octombrie 2018 15:38:08
Problema Numarare triunghiuri Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.78 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
ifstream f("nrtri.in");
ofstream g("nrtri.out");

int n, nrt, v[801];

int cautbin(int val)
{
    int st = 1, dr = n, poz = -1;
    while(st <= dr)
    {
        int med = (st + dr) / 2;
        if(val >= v[med])
        {
            st = med + 1;
            poz = med;
        }
        else
            dr = med - 1;
    }
    return poz;
}

int main()
{
    f >> n;
    for(int i = 1; i <= n; i++)
        f >> v[i];
    sort(v + 1, v + n + 1);
    for(int i = 1; i < n - 1; i++)
        for(int j = i + 1; j < n; j++)
        {
            int poz = cautbin(v[i] + v[j]);
            if(poz != -1)
                nrt += (poz - j);
        }
    g << nrt;
    return 0;
}