Cod sursa(job #2743139)

Utilizator Vivi...Kulik Vivian Vivi... Data 22 aprilie 2021 16:48:43
Problema Numarare triunghiuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.14 kb
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
ifstream in("nrtri.in");
ofstream out("nrtri.out");
int main()
{
    int n,v[801],nr=0,p;
    in>>n;
    for(int i=1; i<=n; i++)
        in>>v[i];
    sort(v+1,v+n+1);
    for(int i=1; i<=n-2; i++)
    {
        for(int j=i+1; j<=n-1; j++)
        {
            int st=j+1;
            int dr=n;
            int s=v[i]+v[j];
            p=0;
            while(st<=dr)
            {
                int mij=(st+dr)/2;
                if(v[mij]==s)
                {
                    p=mij;
                    st++;
                }
                else if(v[mij]>s)
                    dr=mij-1;
                else
                    st=mij+1;
            }
            if(p==0)
                p=dr;
            nr=nr+p-j;
        }
    }
    /*for(int i=1;i<=n-2;i++)
    {
        for(int j=i+1;j<=n-1;j++)
        {
            for(int k=j+1;k<=n;k++)
            {
                if(v[i]+v[j]>=v[k] && v[i]+v[k]>=v[j] && v[j]+v[k]>=v[i])
                    nr++;
            }
        }
    }*/
    out<<nr;
    return 0;
}