Cod sursa(job #1264853)

Utilizator stanamd123Stana Marius Vlad stanamd123 Data 16 noiembrie 2014 13:31:54
Problema Numarare triunghiuri Scor 70
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.22 kb
#include <iostream>
#include <fstream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

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

int N,V[805],i,j,rez,u,p,li,ls,m;

int caut1 (int p, int u, int sum)
{
    for (int i=p; i<=u; i++)
    {
        if (V[i]>=sum)
        {
            return i;
        }
    }
    return u;
}

int caut2 (int p, int u, int sum)
{
    int poz;
    for (int i=1; i<=u; i++)
    {
        if (V[i]<=sum)
        {
            poz=i;
        }
    }
    return poz;
}

int cautBin1 (int p, int u, int sum)
{
    li=p;
    ls=u;
    while (li<=ls)
    {
        m=(li+ls)/2;
        if (V[m]>=sum)
        {
            ls=m-1;
        }
        else
        {
            li=m+1;
        }
    }
    return li;
}



int main()
{
    fin>>N;

    for (i=1; i<=N; i++)
    {
        fin>>V[i];
    }

    sort(V+1,V+N+1);

    for (i=1; i<=N-1; i++)
    {
        for (j=i+1; j<=N; j++)
        {
            u=caut2(1,N,V[i]+V[j]);
            p=cautBin1(1,N,V[j]-V[i]);
            rez += u - p + 1 - (i >= p && i <=u) - (j >= p && j <= u);
        }
    }

    fout<<rez/3;
    return 0;
}