LeetCode 640. 求解方程(C++、python)

求解一个给定的方程,将x以字符串"x=#value"的形式返回。该方程仅包含'+',' - '操作,变量 x 和其对应系数。

如果方程没有解,请返回“No solution”。

如果方程有无限解,则返回“Infinite solutions”。

如果方程中只有一个解,要保证返回值 x 是一个整数。

示例 1:

输入: "x+5-3+x=6+x-2"
输出: "x=2"

示例 2:

输入: "x=x"
输出: "Infinite solutions"

示例 3:

输入: "2x=x"
输出: "x=0"

示例 4:

输入: "2x+3x-6x=x+2"
输出: "x=-1"

示例 5:

输入: "x=x+2"
输出: "No solution"

本题有两个注意点:

1)+x或-x时,注意x的系数是1,容易遗忘

2)当x的系数是0时,直接忽略,比如0x

C++

class Solution {
public:
    string solveEquation(string equation) 
    {
        int num=equation.length();
        int index;
        for(int i=0;i<num;i++)
        {
            if('='==equation[i])
            {
                index=i;
                break;
            }
        }
        if(equation[0]!='+')
        {
            equation.insert(0,"+");
            index++;
            num++;
        }
        if(equation[index+1]!='+')
        {
            equation.insert(index+1,"+");
            num++;
        }
        vector<int> tmp1;
        vector<int> tmp2;
        for(int i=0;i<index;i++)
        {
            if('-'==equation[i] || '+'==equation[i])
            {
                tmp1.push_back(i);
            }
        }
        for(int i=index+1;i<num;i++)
        {
            if('-'==equation[i] || '+'==equation[i])
            {
                tmp2.push_back(i);
            }
        }
        tmp1.push_back(index);
        tmp2.push_back(num);
        int m=tmp1.size();
        int n=tmp2.size();
        int left_num=0;
        int left_x=0;
        int right_num=0;
        int right_x=0;
        for(int i=1;i<m;i++)
        {
            if('x'==equation[tmp1[i]-1])
            {
                string cc=equation.substr(tmp1[i-1]+1,tmp1[i]-tmp1[i-1]-2);
                if("0"!=cc)
                {
                    int t=atoi(cc.c_str());
                    if('+'==equation[tmp1[i-1]])
                    {
                        if(0==t)
                        {
                            t=1;
                        }
                        left_x+=t;
                    }
                    else
                    {
                        if(0==t)
                        {
                            t=1;
                        }
                        left_x-=t;
                    }
                }
            }
            else
            {
                string cc=equation.substr(tmp1[i-1]+1,tmp1[i]-tmp1[i-1]-1);
                int t=atoi(cc.c_str());
                if('+'==equation[tmp1[i-1]])
                {
                    left_num+=t;
                }
                else
                {
                    left_num-=t;
                }
            }
        }
        for(int i=1;i<n;i++)
        {
            if('x'==equation[tmp2[i]-1])
            {
                string cc=equation.substr(tmp2[i-1]+1,tmp2[i]-tmp2[i-1]-2);
                if("0"!=cc)
                {
                    int t=atoi(cc.c_str());
                    if('+'==equation[tmp2[i-1]])
                    {
                        if(0==t)
                        {
                            t=1;
                        }
                        right_x+=t;
                    }
                    else
                    {
                        if(0==t)
                        {
                            t=1;
                        }
                        right_x-=t;
                    }
                }
            }
            else
            {
                string cc=equation.substr(tmp2[i-1]+1,tmp2[i]-tmp2[i-1]-1);
                int t=atoi(cc.c_str());
                if('+'==equation[tmp2[i-1]])
                {
                    right_num+=t;
                }
                else
                {
                    right_num-=t;
                }
            }
        }
        if(left_x==right_x)
        {
            if(left_num==right_num)
            {
                return "Infinite solutions";
            }
            else
            {
                return "No solution";
            }
        }
        else
        {
            int tt=(right_num-left_num)/(left_x-right_x);
            string res="x=";
            return res+to_string(tt);
        }       
    }
};

python

class Solution:
    def solveEquation(self, equation):
        """
        :type equation: str
        :rtype: str
        """
        num=len(equation)
        for i in range(num):
            if '='==equation[i]:
                index=i
        vec=list(equation)
        if '+'!=vec[0]:
            vec.insert(0,'+')
            index+=1
            num+=1
        if '+'!=vec[index+1]:
            vec.insert(index+1,'+')
            num+=1
        vec=''.join(vec)
        tmp1=[]
        tmp2=[]
        for i in range(index):
            if '+'==vec[i] or '-'==vec[i]:
                tmp1.append(i)
        for i in range(index+1,num):
            if '+'==vec[i] or '-'==vec[i]:
                tmp2.append(i)
        tmp1.append(index)
        tmp2.append(num)
        m=len(tmp1)
        n=len(tmp2)
        left_x=0
        left_num=0
        right_x=0
        right_num=0
        for i in range(1,m):
            if 'x'==vec[tmp1[i]-1]:
                cc=vec[tmp1[i-1]+1:tmp1[i]-1]
                if cc!="0":
                    if cc=="":
                        t=1
                    else:
                        t=int(cc)
                    if '+'==vec[tmp1[i-1]]:
                        left_x+=t
                    else:
                        left_x-=t
            else:
                cc=vec[tmp1[i-1]+1:tmp1[i]]
                if cc!="":
                    t=int(cc)
                    if '+'==vec[tmp1[i-1]]:
                        left_num+=t
                    else:
                        left_num-=t
        for i in range(1,n):
            if 'x'==vec[tmp2[i]-1]:
                cc=vec[tmp2[i-1]+1:tmp2[i]-1]
                if cc!="0":
                    if cc=="":
                        t=1
                    else:
                        t=int(cc)
                    if '+'==vec[tmp2[i-1]]:
                        right_x+=t
                    else:
                        right_x-=t
            else:
                cc=vec[tmp2[i-1]+1:tmp2[i]]
                if cc!="":
                    t=int(cc)
                    if '+'==vec[tmp2[i-1]]:
                        right_num+=t
                    else:
                        right_num-=t
        if left_x==right_x:
            if left_num==right_num:
                return "Infinite solutions"
            else:
                return "No solution"
        else:
            res=(right_num-left_num)//(left_x-right_x)
            return "x="+str(res)

 

你好!对于LeetCode上的问题994.腐烂的橘子,你可以使用Python来解决。下面是一个示例代码: ```python from collections import deque def orangesRotting(grid): # 记录网格的行数和列数 row, col = len(grid), len(grid[0]) # 定义四个方向:上、下、左、右 directions = [(-1, 0), (1, 0), (0, -1), (0, 1)] # 使用队列来保存腐烂的橘子的位置 queue = deque() # 记录新鲜橘子的数量 fresh_count = 0 # 遍历整个网格,初始化队列和新鲜橘子的数量 for i in range(row): for j in range(col): if grid[i][j] == 2: # 腐烂的橘子 queue.append((i, j)) elif grid[i][j] == 1: # 新鲜橘子 fresh_count += 1 # 如果新鲜橘子的数量为0,直接返回0 if fresh_count == 0: return 0 # 初始化分钟数 minutes = 0 # 开始进行BFS,直到队列为空 while queue: # 记录当前分钟数下,队列中的元素数量 size = len(queue) # 遍历当前分钟数下的所有腐烂的橘子 for _ in range(size): x, y = queue.popleft() # 遍历四个方向 for dx, dy in directions: nx, ny = x + dx, y + dy # 判断新位置是否在网格内,并且是新鲜橘子 if 0 <= nx < row and 0 <= ny < col and grid[nx][ny] == 1: # 将新鲜橘子变为腐烂状态 grid[nx][ny] = 2 # 将新鲜橘子的位置加入队列 queue.append((nx, ny)) # 新鲜橘子的数量减1 fresh_count -= 1 # 如果当前分钟数下,没有新鲜橘子了,结束循环 if fresh_count == 0: break # 每遍历完一层,分钟数加1 minutes += 1 # 如果最后还有新鲜橘子,返回-1,否则返回分钟数 return -1 if fresh_count > 0 else minutes ``` 你可以将给定的网格作为参数传递给`orangesRotting`函数来测试它。请注意,该代码使用了BFS算法来遍历橘子,并计算腐烂的分钟数。希望能对你有所帮助!如果有任何疑问,请随时问我。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值