Cod sursa(job #2078754)

Utilizator MrRobotMrRobot MrRobot Data 29 noiembrie 2017 21:47:54
Problema Trapez Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.04 kb
#include <iostream>
#include <fstream>
#include <stdlib.h>

using namespace std;

struct {
    long long x, y;
} p[1000], m[1000000];

///p: A(x, y)
///m: dif de y/ dif de x => m[].x = dif de y
///                         m[].y = dif de x

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

long long cmmdc(long long a, long long b)
{
    if( b != 0 )
        return cmmdc(b, a%b);
    else
        return a;
}

int part_Hoare ( int s, int d )
{

    int i, j;
        i = s-1;
        j= d+1;
    int randpoz, r1, r2, r3;
    int nr = d-s+1;

    r1 = rand() % nr + s;
    r2 = rand() % nr + s;
    r3 = rand() % nr + s;

    int minr, maxr;
    minr = min ( min(r1, r2), min(r2, r3) );
    maxr = max ( max(r1, r2), max(r2, r3) );

    randpoz = r1+r2+r3 - minr-maxr;

    int pivot_x = m[randpoz].x;
    int pivot_y = m[randpoz].y;

    while (true){
        do
        {
            ++i;
        }while( m[i].x * pivot_y < m[i].y * pivot_x );
        do
        {
            --j;
        }while( m[j].x * pivot_y > m[j].y * pivot_x);
        if( i>=j )
            return j;
        //swap
        int aux_x, aux_y;
        aux_x = m[i].x;
        aux_y = m[i].y;
        m[i].x = m[j].x;
        m[i].y = m[j].y;
        m[j].x = aux_x;
        m[j].y = aux_y;
    }
}

void qsort (int i, int j)
{
    if(i<j){
        int k = part_Hoare( i, j );
        qsort(i, k);
        qsort(k+1, j);
    }
}

/*void sortare(int k)
{
    int i, j;
    for(i=0; i<k; i++)
        for(j=i+1; j<=k; j++)
        {
            if( (m[i].x * m[j].y) > (m[i].y * m[j].x) )
            {
                int aux_x, aux_y;
                aux_x = m[i].x;
                aux_y = m[i].y;
                m[i].x = m[j].x;
                m[i].y = m[j].y;
                m[j].x = aux_x;
                m[j].y = aux_y;
            }
        }
}*/

int main()
{
    int n, i, j;
    long long nr_part=0, nr_total=0, nr_x0=0;
    fin>>n;
    int k = -1;

    for( i = 0; i < n ; i++ )
        {
            fin >> p[i].x >> p[i].y;
            for( j = 0; j < i; j++ )
                {
                    if(p[i].x == p[j].x)
                        nr_x0++;
                    else
                    {
                        ++k;
                        m[k].x = p[j].y - p[i].y;
                        m[k].y = p[j].x - p[i].x;
                        int d = cmmdc( abs(m[k].x), abs(m[k].y) );
                        m[k].x /= d;
                        m[k].y /=d;
                    }
                }
        }




    qsort(0, k);




    nr_x0 = nr_x0*(nr_x0-1)/2;
    nr_total += nr_x0;

    i=1;
    while( i <= k )
    {
        nr_part=1;
        //while( m[i].x * m[i-1].y == m[i].y * m[i-1].x )
        while( ( m[i-1].x == m[i].x ) && ( m[i-1].y == m[i].y ) )
        {
            nr_part++;
            i++;
        }

        nr_part = nr_part*(nr_part-1)/2;

        nr_total += nr_part;
        i++;
    }
    fout<<nr_total;
    return 0;
}