Cod sursa(job #2609778)

Utilizator Stefan_RaduStefan Radu Stefan_Radu Data 3 mai 2020 15:26:45
Problema Numarare triunghiuri Scor 70
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
// By Stefan Radu

#include <algorithm>
#include <fstream>
#include <iomanip>
#include <cassert>
#include <vector>
#include <string>
#include <cctype>
#include <queue>
#include <deque>
#include <cmath>
#include <stack>
#include <map>
#include <set>

using namespace std;

ifstream cin("nrtri.in");
ofstream cout("nrtri.out");

#define sz(x) (int)(x).size()

typedef pair < int, int > pii;
typedef long long ll;
typedef long double ld;
typedef unsigned int ui;
typedef unsigned long long ull;

int main() {

#ifdef STEF
  freopen("input", "r", stdin);
  freopen("output", "w", stdout);
#endif

  ios::sync_with_stdio(false);
  cin.tie(0);cout.tie(0);

  int n; cin >> n;

  multiset < int > s;
  vector < int > lengths(n);
  for (int i = 0; i < n; ++ i) {
    cin >> lengths[i];
    s.insert(lengths[i]);
  }

  int cnt = 0;
  for (int i = 0; i < n; ++ i) {
    s.erase(s.find(lengths[i]));
    for (int j = i + 1; j < n; ++ j) {
      s.erase(s.find(lengths[j]));
      auto a = s.lower_bound(abs(lengths[i] - lengths[j]));
      auto b = s.upper_bound(lengths[i] + lengths[j]);
      cnt +=  distance(a, b);
      s.insert(lengths[j]);
    }
    s.insert(lengths[i]);
  }

  cout << cnt / 3 << '\n';
}