Cod sursa(job #397884)

Utilizator nandoLicker Nandor nando Data 17 februarie 2010 17:16:47
Problema Ecuatie Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.68 kb
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>

#define MAX 1000000

int sol[MAX][4],sol2[MAX][4],ns=0,ns2=0;
int a,b,c,i,k,c1,c2,d1,d2;
long long d;

void printeq(int eq[]){
	printf("(");
	if(eq[0]!=1&&eq[0]!=-1){
		printf("%d",eq[0]);
	}else if(eq[0]==-1){
		printf("-");
	}
	printf("x%c%u)(",(eq[1]>0)?'+':'-',(eq[1]>0)?eq[1]:-eq[1]);
	if(eq[2]!=1&&eq[2]!=-1){
		printf("%d",eq[2]);
	}else if(eq[2]==-1){
		printf("-");
	}
	printf("x%c%u)\n",(eq[3]>0)?'+':'-',(eq[3]>0)?eq[3]:-eq[3]);
}
void addSol(int s){
	sol[ns][0]=d1*s;
	sol[ns][1]=-c1*s;
	sol[ns][2]=d2*(a/s);
	sol[ns][3]=-c2*(a/s);
	ns++;
	sol[ns][0]=d2*s;
	sol[ns][1]=-c2*s;
	sol[ns][2]=d1*(a/s);
	sol[ns][3]=-c1*(a/s);
	ns++;
}
void reduce(int& a,int& b){
	int a0=a,b0=b,r;
	while(b0){
		r=a0%b0;
		a0=b0;
		b0=r;
	}
	a/=a0,b/=a0;
}
int comp(const void* p1,const void* p2){
	int *v1=(int*)p1,*v2=(int*)p2;
	if(v1[0]-v2[0]==0){
		return v1[1]-v2[1];
	}else{
		return v1[0]-v2[0];
	}
}
int main(){

	FILE* fin=fopen("ecuatie.in","r");
	freopen("ecuatie.out","wt",stdout);
	
	fscanf(fin,"%d %d %d %d",&a,&b,&c,&k);
	d=b*b-4*a*c;
	i=sqrt(abs(d));
	if(i*i==d){
		c1=-b-i,c2=-b+i,d1=d2=2*a;
		reduce(c1,d1);
		reduce(c2,d2);
		a/=(d1*d2);
		int l=sqrt(abs(a));
		for(int i=1;i<=l;i++){
			if(a%i==0){
				addSol(i),addSol(-i);
				addSol(a/i),addSol(-a/i);
			}
		}
		qsort(sol,ns,sizeof(int)*4,comp);
		for(int i=0;i<ns;i++){
			memcpy(sol2[ns2++],sol[i],16);
			while(!memcmp(sol[i],sol[i+1],16)&&i<ns-1){
				i++;
			}
		}
		if(k<=ns2){
			printeq(sol2[k-1]);
		}else{
			printf("-1");
		}
	}else{
		printf("-1");
	}
getchar();
	return 0;
}