365天挑战LeetCode1000题——Day 053 求解方程解析模拟

最新推荐文章于 2025-11-25 00:06:26 发布

原创最新推荐文章于 2025-11-25 00:06:26 发布 · 202 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++

LeetCode千题之路专栏收录该内容

124 篇文章

订阅专栏

本文介绍了一种解析一元方程的方法，通过解析字符串形式的数学表达式来获取方程中未知数的系数和常数项，进而求解方程。代码实现了对方程左右两边分别解析并计算未知数系数和常数项的功能。

640. 求解方程

在这里插入图片描述

代码实现（自解）

class Solution {
private:
    void parse(string str, int& x, int& co) {
        int pstr = 0;
        int n = str.size();
        bool flag = false;
        string tmp = "";
        int val = 0;
        x = 0, co = 0;
        while (pstr != n) {
            if (str[pstr] == '-') {
                flag = true;
                pstr++;
            }
            else if (str[pstr] == '+') {
                pstr++;
            }
            while (str[pstr] >= '0' && str[pstr] <= '9') {
                tmp += str[pstr++];
            }
            if (str[pstr] == 'x') {
                if (tmp == "") val = 1;
                else val = stoi(tmp);
                if (flag) x -= val;
                else x += val;
                
            }
            else {
                if (flag) co -= stoi(tmp);
                else co += stoi(tmp);
                pstr--;
            }

            pstr++;
            flag = false;
            tmp = "";
        }
    }
public:
    string solveEquation(string equation) {
        int equal = equation.find('=', 0);
        string left = equation.substr(0, equal);
        string right = equation.substr(equal + 1);
        int x1, co1, x2, co2;
        parse(left, x1, co1);
        parse(right, x2, co2);
        // cout << "x1 : " << x1 << ", co1 : " << co1 << endl;
        // cout << "x2 : " << x2 << ", co2 : " << co2 << endl;
        if (x1 == x2 && co1 == co2) return "Infinite solutions";
        if (x1 == x2 && co1 != co2) return "No solution";
        return "x=" + to_string((co2 - co1) / (x1 - x2));
    }
};