Cod sursa(job #995032)

Utilizator chiriacandrei25Chiriac Andrei chiriacandrei25 Data 7 septembrie 2013 00:22:04
Problema Ecuatie Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.35 kb
#include <fstream>
#include <cmath>
#include <algorithm>
using namespace std;

#define lint long long

ifstream f("ecuatie.in");
ofstream g("ecuatie.out");

struct vector
{
    lint p1,q1,p2,q2;
} q[5000];

lint A,B,C,K,l,delta;
long double X1,X2;
bool stop;

struct sorting1
{
    bool operator () (const vector& lhs , const vector& rhs)
    {
        if (lhs.p1 == rhs.p1)
            return lhs.q1 == rhs.q1;

        return lhs.p1 == rhs.p1;
    }
};
struct sorting
{
    bool operator () (const vector& lhs , const vector& rhs)
    {
        if (lhs.p1 == rhs.p1)
            return lhs.q1 < rhs.q1;

        return lhs.p1 < rhs.p1;
    }
};
inline void calcul()
{
    delta=B*B-(A*C<<2);
    if (sqrt(delta)==(lint)sqrt(delta)) delta=(lint)sqrt(delta);
    else stop=1;
    X1=(long double)((-B-delta)/(A*2.0));
    X2=(long double)((-B+delta)/(A*2.0));
}
inline void verf(lint d)
{
    long double a,b;
    a=-X1*d;
    b=-X2*(A/d);
    if ((a==(lint)(a)) && (b==(lint)(b)))
    {
        q[l].q1=(lint)(a);
        q[l].p1=d;
        q[l].q2=(lint)(b);
        q[l++].p2=A/d;
    }
    a=-X2*d;
    b=-X1*(A/d);
    if ((a==(lint)(a)) && (b==(lint)(b)))
    {
        q[l].q1=(lint)(a);
        q[l].p1=d;
        q[l].q2=(lint)(b);
        q[l++].p2=A/d;
    }
}
inline void divizor()
{
    lint k=abs(A),i,cnt,d;
    for ( i = 1, cnt = (1<<16) ; cnt; cnt >>=1)
        if ((i + cnt) * (i + cnt) <= k) i += cnt;
    k=i;
    for ( d = 1; d <= k; d++)
        if (A%d==0)
        {
            verf(d);
            verf(-d);
            verf(A/d);
            verf(-A/d);
        }
}
int main()
{
    f>>A>>B>>C>>K;
    calcul();
    if (stop) g<<"-1";
    else
    {
        divizor();
        sort(q,q + l,sorting());
        l = unique(q, q + l ,sorting1()) - q;
        if (K>l) g<<"-1";
        else
        {
            g<<"(";
            if (q[K-1].p1==-1) g<<"-x";
            else if (q[K-1].p1==1) g<<"x";
            else g<<q[K-1].p1<<"x";
            if (q[K-1].q1>=0) g<<"+"<<q[K-1].q1<<")(";
            else g<<q[K-1].q1<<")(";
            if (q[K-1].p2==-1) g<<"-x";
            else if (q[K-1].p2==1) g<<"x";
            else g<<q[K-1].p2<<"x";
            if (q[K-1].q2>=0) g<<"+"<<q[K-1].q2<<")";
            else g<<q[K-1].q2<<")";
        }
    }
    return 0;
}