Cod sursa(job #1507333)

Utilizator RaduMirceaAndreiRadu Mircea Andrei RaduMirceaAndrei Data 21 octombrie 2015 16:51:18
Problema Numarare triunghiuri Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
# include <fstream>
# include <algorithm>
using namespace std;
ifstream fin("nrtri.in");
ofstream fout("nrtri.out");
int n,v[801],i,k,j,sol,st,dr;
int main () {
    fin>>n;
    for(i=1;i<=n;i++){
        fin>>v[i];
    }
    k=3;
    sort(v+1, v+n+1);
    for(i=1;i<=n-2;i++){
        for(j=i+1;j<=n-1;j++){
            //dela pozitia j+1 pana la n, primele cateva numere
            //pot fi a treia latura a triunghiului cu primele 2 v[i] si v[j]
            // si apoi nu mai poate fi niciuna
            //cautam binar intre pozitiile j+1 si n cea mai din dreapta pozitie
            //<= v[i]+v[j]
            st = j+1; dr = n;
            while (st <= dr) {
                int mid = (st + dr)/2;
                if (v[mid] <= v[i]+v[j] )
                    st = mid+1;
                else
                    dr = mid-1;
            }
            sol += st-(j+1);
        }
    }
    fout<<sol<<" ";
    return 0;
}