Cod sursa(job #2655694)

Utilizator Cosmin2004_InfoMoldoveanu Cosmin Cosmin2004_Info Data 5 octombrie 2020 11:21:52
Problema Numarare triunghiuri Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <cstdio>
#include <algorithm>

using namespace std;
int v[1005];
int bs_right(int st, int dr, int x) {
    int med, last = -1;
    while(st <= dr) {
        med = (st + dr) >> 1;
        if(x >= v[med]) {
            last = med;
            st = med + 1;
        } else dr = med - 1;
    }
    return last;
}

int main()
{
    freopen("nrtri.in", "r", stdin);
    freopen("nrtri.out", "w", stdout);
    int n, s = 0, bsr;
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) scanf("%d", &v[i]);
    sort(v + 1, v + n + 1);
    for(int i = 1; i < n - 1; i++) {
        for(int j = i + 1; j < n; j++) {
            bsr = bs_right(j + 1, n, v[i] + v[j]);
            if(bsr != -1)
                s += bsr - j;
        }
    }
    printf("%d", s);
    return 0;
}