【360秋招笔试】编程题第二题：修改Web（C++的AC解法）

题目

先看样例：

6
16=1+2*3
7*8*9=54
1+1=1+22
4*6=22+2
15+7=1+2
11+1=1+5

n表示输入n行数据，下面每一行数据表示一个等式。如果能满足 在等式中添加任意一个数字 使得等式两边成立，则输出Yes，否则输出No。如果等式本来就相等，也输出Yes。

符号只有+，*和=。都是正整数。

想测试自己的拆分或计算是否正确，可以只使用一个样例：

1
16=1+2*3

思路

过程略微复杂的模拟题。

想要进行计算，需要将等式（字符串）拆分为数字和字符。计算过程类似于逆波兰表达式，用栈来实现。至于是否存在添加一个数字使得等式成立，我们可以枚举。

本题解C++。（~~这种数据输入感觉用C++会方便，如果用JS需要异步？？360的编程题竟然没有给输入的模板，直接ACM模式~~ ）

输入字符串，得到数字和字符的函数：其中cntNum、cntCh等是指针，用来记录一共拆分出了多少数。

void stringTo(string a){int temp=0;for(int i=0;a[i];i++){if(a[i]>='0'&&a[i]<='9'){temp*=10;temp+=a[i]-'0';}else{ch[cntCh]=a[i];cntCh++;num[cntNum]=temp;cntNum++;temp=0;}}num[cntNum]=temp;cntNum++;temp=0;
}

对等号=左右两边的数字进行计算，思想是逆波兰表达式。

这里的参数left、right表示计算数组中[left,right]的答案。这样，只需要改变参数就可以分别计算等号左边和等号右边。

注意，数字的数量会比符号多1，因此可以先把第一个数字压入栈。

//计算左右两边的大小 
//左 1,indexx
//右 indexx+1,end 
int cal(int left,int right){if(left==right) return num[left];stack<int>numm;stack<char>chh;numm.push(num[left]);
//	数字 for(int i=left+1;i<=right;i++){//对应的操作符是i-1的下标 char op=ch[i-1];//优先级高 if(op=='*'){int qian=numm.top();numm.pop();int hou=num[i];int temp=qian*hou;numm.push(temp);}else{numm.push(num[i]);chh.push(ch[i-1]);}}//	计算加减while(chh.size()){char op=chh.top();chh.pop();int hou=numm.top();numm.pop();int qian=numm.top();numm.pop();int temp=qian+hou;numm.push(temp);} return numm.top();
}

枚举在字符串中每一个位置都插入一个数字并计算：

for(int i=0;a[i];i++){c.clear();b+=a[i];c+=b;for(int j=0;j<=9;j++){c+=char('0'+j);c+=a.substr(i+1);if(solve(c)) {flag=1;break;}c.clear();c+=b;}if(flag) break;
}

总体的过程：将字符串转换为数字和符号、计算。这个过程可以封装一下：

bool solve(string a){clearr();stringTo(a);findEqual();l=cal(0,indexx);r=cal(indexx+1,cntNum-1);if(l==r) return true;else return false;
}

总体代码（AC）

#include<bits/stdc++.h>
using namespace std;
int t;string a;
int num[5000];
char ch[5000];
int cntNum=0,cntCh=0;
int indexx;//=的下标 
int l,r;
void clearr(){cntNum=0,cntCh=0;
}
void stringTo(string a){int temp=0;for(int i=0;a[i];i++){if(a[i]>='0'&&a[i]<='9'){temp*=10;temp+=a[i]-'0';}else{ch[cntCh]=a[i];cntCh++;num[cntNum]=temp;cntNum++;temp=0;}}num[cntNum]=temp;cntNum++;temp=0;
}
//找到=的下标index 对应num的下标，index及其之前的都是=号左边的 
void findEqual(){for(int i=0;i<cntCh;i++){if(ch[i]=='='){indexx=i;break;}} 
}
//计算左右两边的大小 
//左 1,indexx
//右 indexx+1,end 
int cal(int left,int right){if(left==right) return num[left];stack<int>numm;stack<char>chh;numm.push(num[left]);
//	数字 for(int i=left+1;i<=right;i++){//对应的操作符是i-1的下标 char op=ch[i-1];//优先级高 if(op=='*'){int qian=numm.top();numm.pop();int hou=num[i];int temp=qian*hou;numm.push(temp);}else{numm.push(num[i]);chh.push(ch[i-1]);}}//	计算加减while(chh.size()){char op=chh.top();chh.pop();int hou=numm.top();numm.pop();int qian=numm.top();numm.pop();int temp=qian+hou;numm.push(temp);} return numm.top();
}bool solve(string a){clearr();stringTo(a);findEqual();l=cal(0,indexx);r=cal(indexx+1,cntNum-1);if(l==r) return true;else return false;
}int main(){cin>>t;while(t--){clearr();l=0,r=0;cin>>a;stringTo(a);//        找= indexfindEqual();l=cal(0,indexx);r=cal(indexx+1,cntNum-1);if(l==r) {cout<<"Yes"<<endl;continue;}string b,c;int flag=0;//在最前面加for(int i=0;i<=9;i++){c.clear();c+=char('0'+i);c+=a;if(solve(c)) {flag=1;break;}}if(flag) {cout<<"Yes"<<endl;continue;}b.clear();c.clear();for(int i=0;a[i];i++){c.clear();b+=a[i];c+=b;for(int j=0;j<=9;j++){c+=char('0'+j);c+=a.substr(i+1);if(solve(c)) {flag=1;break;}c.clear();c+=b;}if(flag) break;}if(flag) {cout<<"Yes"<<endl;continue;}else cout<<"No"<<endl;    }return 0;
}