基于二次方程构建含复数解的求解函数的技术实现问询

嘿，我看你正在折腾一个能处理复数解的二次方程求解函数，思路是对的，但目前的代码有几个关键小问题，我来帮你捋顺并修正：

首先，你当前用std::vector<float>作为返回类型，这根本存不了复数解啊——复数需要同时存储实部和虚部，得把返回类型改成std::vector<std::complex<float>>才行。

然后是根的计算逻辑，你写的x1 = (-b)/2*a完全踩了运算优先级的坑！除法和乘法是左结合的，这行代码等价于((-b)/2)*a，和二次方程根的公式[-b ± √D]/(2a)完全不符，必须给2*a加括号，保证先算分母。

还有判别式的处理：当判别式为负时，你只是把它转成了复数，但没计算它的平方根虚部。其实不用分开写实根和复根的逻辑，C++的std::complex已经支持对复数的平方根运算，我们可以统一处理所有情况——不管判别式是正还是负，直接用复数类型计算就行，代码会自动处理实部和虚部。

给你一个修正后的完整实现：

#include <vector>
#include <complex>
#include <cmath>

std::vector<std::complex<float>> quadratic_formula(float a, float b, float c) {
    std::vector<std::complex<float>> result;
    
    // 先把判别式转成复数类型，这样不管正负都能正确计算平方根
    std::complex<float> discriminant = std::complex<float>(b*b - 4*a*c);
    std::complex<float> sqrt_disc = std::sqrt(discriminant);
    // 分母是2a，用复数类型存储（其实这里是实数，但统一类型更安全）
    std::complex<float> denominator = 2 * a;

    // 直接套用二次方程求根公式，复数运算会自动处理实部和虚部
    std::complex<float> root1 = (-b + sqrt_disc) / denominator;
    std::complex<float> root2 = (-b - sqrt_disc) / denominator;

    result.push_back(root1);
    result.push_back(root2);
    return result;
}

你可以用这段测试代码验证一下：

#include <iostream>

int main() {
    // 测试方程 x² + 2x +5 =0，预期解是 -1±2i
    auto roots = quadratic_formula(1, 2, 5);
    std::cout << "第一个根：" << roots[0] << std::endl;
    std::cout << "第二个根：" << roots[1] << std::endl;
    return 0;
}

另外给你两个小提示：

• 记得加判断处理a=0的情况，这时候方程退化为一次方程bx +c =0，不能用二次方程的公式
• 如果对精度要求高，可以把float换成double，std::complex<double>的精度会更好

内容来源于stack exchange