Cod sursa(job #343496)

Utilizator ProtomanAndrei Purice Protoman Data 26 august 2009 00:03:00
Problema Ecuatie Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.92 kb
#include <algorithm>
#include <iostream>
#include <vector>
#include <string>
#include <math.h>

#define ll long long
#define pb push_back
#define mp make_pair
#define f first
#define s second

using namespace std;

ll a, b, c, k, x1, x2;
vector <ll> vctDiviz;
vector <pair <pair <ll, ll>, pair <ll, ll> > > vctSol;

inline ll llabs(ll x)
{
	if (x < 0)
		return -x;
	return x;
}

inline ll gcd(ll a, ll b)
{
	if (!b)
		return a;
	return gcd(b, a % b);
}

inline void form(ll coef, ll af)
{
	printf("(");
	if (coef < 0 && coef == -1)
		printf("-");
	if (coef != 1 && coef != -1)
		printf("%d", coef);
	printf("x");

	if (af > 0)
		printf("+");
	printf("%d)", af);
}

int main()
{
	freopen("ecuatie.in", "r", stdin);
	freopen("ecuatie.out", "w", stdout);

	scanf("%d %d %d %d", &a, &b, &c, &k);
	k--;

	ll cmmdc = llabs(gcd(llabs(a), gcd(llabs(b), llabs(c))));

	for (int diviz = 1; diviz * diviz <= cmmdc; diviz++)
		if (cmmdc % diviz == 0)
		{
			vctDiviz.pb(diviz);
			vctDiviz.pb(-diviz);
			
			if (diviz * diviz != cmmdc)
			{
				vctDiviz.pb(cmmdc / diviz);
				vctDiviz.pb(-cmmdc / diviz);
			}
		}

	for (int i = 0; i < vctDiviz.size(); i++)
	{
		ll p1 = (ll) -b - sqrt((double) b * b - 4 * a * c);
		ll p2 = (ll) -b + sqrt((double) b * b - 4 * a * c);
		vctSol.pb(mp(mp(vctDiviz[i] * (2 * a) / gcd(p1, 2 * a), vctDiviz[i] * p1 / gcd(p1, 2 * a)), 
			mp(a / (vctDiviz[i] * (2 * a) / gcd(p1, 2 * a)), c / (vctDiviz[i] * p1 / gcd(p1, 2 * a)))));
		vctSol.pb(mp(mp(vctDiviz[i] * (2 * a) / gcd(p2, 2 * a), vctDiviz[i] * p2 / gcd(p2, 2 * a)), 
			mp(a / (vctDiviz[i] * (2 * a) / gcd(p2, 2 * a)), c / (vctDiviz[i] * p2 / gcd(p2, 2 * a)))));
	}

	sort(vctSol.begin(), vctSol.end());

	if (k >= vctSol.size())
		printf("-1\n");
	else
	{
		form(vctSol[k].f.f, vctSol[k].f.s);
		form(vctSol[k].s.f, vctSol[k].s.s);
		printf("\n");
	}

	fclose(stdin);
	fclose(stdout);
	return 0;
}