Cod sursa(job #2067904)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 16 noiembrie 2017 22:39:30
Problema Trapez Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.26 kb
#include <iostream>
#include <fstream>
#include <iomanip>
#include <cmath>
#include <vector>
#include <queue>
#include <algorithm>

#if 1
#define pv(x) cout<<#x<<" = "<<x<<"; ";cout.flush()
#define pn cout<<endl
#else
#define pv(x)
#define pn
#endif

using namespace std;
ifstream in("trapez.in");
ofstream out("trapez.out");

#define ll long long
#define ull unsigned long long
#define pb push_back
#define mp make_pair
const int NMax = 1e3 + 50;
const int sMax = 1e6 + 5;

int N,nrPanta;
pair<int,int> p[NMax],panta[NMax*NMax];
// p[i] - coordonatele puncutlui i din input;
// panta[i] - o fractie ireductibila care descrie o panta de dreapta;

int euclid(int,int);
// calculeaza cel mai mare divizor comun intre doua numere;

int main() {
    in>>N;

    for (int i=1;i <= N;++i) {
        in>>p[i].first>>p[i].second;

        for (int j=1;j < i;++j) {
            int dy = p[i].second - p[j].second;
            int dx = p[i].first - p[j].first;

            // daca dreapta este verticala, aceasta se trateaza special;
            if (dx == 0) {
                panta[++nrPanta] = mp(0,0);
                continue;
            }

            // fractia se face ireductibila;
            int gcd = euclid(abs(dx),abs(dy));
            dy /= gcd;
            dx /= gcd;

            // numaratorul si numitorul se prelucreaza pentru a identifica unic fiecare panta;
            if (dx < 0 && dy < 0) {
                dx *= -1;
                dy *= -1;
            }
            else if (dx < 0 && dy > 0) {
                dy *= -1;
                dx *= -1;
            }

            // se adauga intr-un vector;
            panta[++nrPanta] = mp(dx,dy);
        }
    }

    sort(panta+1,panta+nrPanta+1);

    int nr = 0,ans = 0;
    // se numara aparitiile fiecarei pante (nr);
    // acestea vor contribui la raspuns cu Combinari de nr luate cate 2,
    // deoarece fiecare doua drepte paralele vor descrie cate un trapez;
    for (int i=1;i <= nrPanta;++i) {
        if (panta[i] == panta[i-1]) {
            ++nr;
        }
        else {
            ans += nr*(nr-1)/2;
            nr = 1;
        }
    }

    out<<ans<<'\n';

    return 0;
}

int euclid(int a,int b) {
    return !b ? a : euclid(b,a%b);
}