Cod sursa(job #2353268)

Utilizator MateiAruxandeiMateiStefan MateiAruxandei Data 24 februarie 2019 09:37:06
Problema Numarare triunghiuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <fstream>
#include <algorithm>

using namespace std;

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

int v[805];

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

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

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

    int rez = 0;
    for(int i = 1; i <= n; ++i)
        for(int j = i + 1; j <= n; ++j) {
            int st = 1;
            int dr = n;
            int poz = -1;
            while(st <= dr) {
                int mij = (st + dr) / 2;

                if(v[mij] <= v[i] + v[j]){
                    poz = mij;
                    st = mij + 1;
                }
                else dr = mij - 1;
            }

            st = 1;
            dr = n;
            int poz2 = -1;
            while(st <= dr) {
                int mij = (st + dr) / 2;

                if(v[mij] >= v[j] - v[i]){
                    poz2 = mij;
                    dr = mij - 1;
                }
                else st = mij + 1;
            }

            if(poz != -1 && poz2 != -1){
                rez += (poz - poz2 + 1);
                if(i <= poz && i >= poz2)
                    --rez;
                if(j <= poz && j >= poz2)
                    --rez;
            }
            else if(poz != -1){
                rez += poz;
                if(i <= poz)
                    --rez;
                if(j <= poz)
                    --rez;
            }

        }
    fout << rez / 3 << '\n';
    return 0;
}