Cod sursa(job #2781414)

Utilizator Johnny07Savu Ioan-Daniel Johnny07 Data 9 octombrie 2021 13:50:08
Problema Numarare triunghiuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <bits/stdc++.h>
using namespace std;

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

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

    sort(v + 1, v + n + 1);

    int cate_triunghiuri = 0;

    for (int i = 1; i <= n; i++)
    {
        for (int j = i + 1; j<= n; j++)
        {
            int sum = v[i] + v[j];
            //vrem sa gasim pozitia k a.i. v[k] <= sum si v[k+1] > sum
            //stim astfel ca toate elementele de la pozitia j pana la k
            // vor fi valide pentru a face un triunghi 
            int st, dr, mij;
            st = j;
            dr = n;
            while (st <= dr)
            {
                mij = (st + dr) / 2;
                if (v[mij] <= sum) 
                    st = mij  + 1;
                else 
                    dr = mij - 1;
            }
            mij = st + dr;
            mij /= 2;
            if (v[mij] > sum)
                mij--;
            cate_triunghiuri += mij - j;

        }
    }

    fout << cate_triunghiuri;
    return 0;
}