用c++的类写一个输出一元二次方程的根的代码

#include <stdio.h>
int main()
{
double a, b, c, discriminant, root1, root2;

printf("输入一元二次方程的系数a, b and c: ");
scanf("%lf %lf %lf",&a, &b, &c);

discriminant = b*b-4*a*c;

// 通过判断判别式的值来求解根
if (discriminant > 0)
{
root1 = (-b+sqrt(discriminant))/(2*a);
root2 = (-b-sqrt(discriminant))/(2*a);

printf("两个不等的实根是 %.2lf 和 %.2lf",root1 , root2);
}
else if (discriminant == 0)
{
root1 = root2 = -b/(2*a);

printf("两个相等的实根是 %.2lf 和 %.2lf", root1, root2);
}
else
{
realPart = -b/(2*a);
imagPart = sqrt(-discriminant)/(2*a);
printf("两个共轭复根是 %.2lf + %.2lfi 和 %.2f - %.2fi", realPart, imagPart, realPart, imagPart);
}

return 0;
}