问题
给定一个字符串str,str表示一个公式
公式里可能有整数,加减乘除,左右括号等符号
返回公式计算结果,难点在于括号可能有很多层?
示例
str= 48*((70-65)-43)+8*1 返回结果 -1816
思想
利用栈将元素加到栈中,栈的起始从左括号开始,到又括号停,不包括括号,没有括号时遇到左括号结束,然后括号中的内用再建立新栈,并优先处理。
代码
public class Code01_ExpressionCompute {
public static int computenum(String str) {
return process(str.toCharArray(),0)[0];
}
private static int[] process(char[] charArray, int i) {
LinkedList<String> que = new LinkedList<String>();
int num = 0;
int[] bar = null;
while(i<charArray.length&&charArray[i]!=')') {
if(charArray[i]>='0'&&charArray[i]<='9') {
num = num*10 +charArray[i++]-'0';
}else if(charArray[i]!='(') {
addnum(que,num);
que.addLast(String.valueOf(charArray[i++]));
num = 0;
}else {
bar = process(charArray, i+1);
num = bar[0];
i = bar[1]+1;
}
}
addnum(que, num);
return new int[] {getnum(que),i};
}
private static void addnum(LinkedList<String> que, int num) {
if(!que.isEmpty()) {
String top = que.pollLast();
int cur = 0;
if(top.equals("+")||top.equals("-")) {
que.addLast(top);
}else {
cur = Integer.valueOf(que.pollLast());
num = top.equals("*")?2cur*num:(cur/num);
}
}
que.addLast(String.valueOf(num));
}
private static int getnum(LinkedList<String> que) {
int res = 0;
int num = 0;
String cur = null;
boolean add = true;
while(!que.isEmpty()) {
cur = que.pollFirst();
if(cur.equals("+")) add = true;
else if(cur.equals("-")) add = false;
else {
num = Integer.valueOf(cur);
res += add? num:(-num);
}
}
return res;
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while(scanner.hasNext()) {
String str = scanner.nextLine();
int ans = computenum(str);
System.out.println(ans);
}
}
}