实验一 算法设计基础 例题五

题目：
假设要计算A×B×C×D×E，其中A、B、C、D和E是矩阵。由于矩阵相乘具有结合性，所以矩阵相乘的顺序是任意的。但是，矩阵相乘时做乘法的次数，取决于所选择的矩阵相乘的顺序。
例如，矩阵A：50×10，B：10×20，C：20×5。
计算A×B×C有两种顺序，即（A×B）×C和A×（B×C）。
第一种顺序需要做乘法15000次，而第二种只要3500次。
编程任务：对给定的矩阵相乘顺序，计算矩阵相乘时所需的乘法次数。
输入格式
输入分为两部分：矩阵列表和矩阵相乘的表达式列表。
第一行是一个整数n（1≤n≤26），表示矩阵数目。接着有n行，每行的开头是一个大写字母，是矩阵的名称，然后是两个整数，表示该矩阵的行数与列数。
输出格式
对第二部分的每个表达式，输出一行：根据表达式中矩阵相乘的顺序，如果相乘时矩阵不匹配，输出“error”，否则输出矩阵相乘时做乘法的次数。

计算过程分为三种情况：
1）左括号
左括号不影响计算，直接跨越过去。
2）右括号
弹出栈顶的两个矩阵，累计这两个矩阵相乘的次数，并将新矩阵的行列数压入堆栈。如果这两个矩阵不能相乘，则输出错误信息“error”，并结束该表达式的计算。
3）矩阵
将该矩阵压入堆栈。
可参考Example1_2.java代码。
Example1_2.java文件：

import java.util.Stack;
import java.util.Scanner;
public class Example1_2 {
	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);		
		int n = scan.nextInt();
                char[] s = new char[n];
		int[][] matrix = new int[n][2];
		for (int i = 0; i < n; i++) {
                        String str=scan.next();
                        s[i]=str.charAt(0);
			matrix[i][0] = scan.nextInt();
			matrix[i][1] = scan.nextInt();
		}
                
		while(scan.hasNext()){
                        String expression = scan.next();
			int result = calculate(matrix, expression);
			System.out.println(result);
		}
		scan.close();
	}

	private static int calculate(int[][] matrix, String expression) {

		int result = 0;
		Stack<Integer> stack = new Stack<Integer>();
		for (int i = 0; i < expression.length(); i++) {
	            {   char item = expression.charAt(i);
                        if (item=='(')
                                continue;
                        if (item==')')
                        {
                            int row1,col1,row2,col2;
                            col2=stack.peek();
                            stack.pop();
                            row2=stack.peek();
                            stack.pop();
                            col1=stack.peek();
                            stack.pop();
                            row1=stack.peek();
                            stack.pop();
                            if (col1!=row2)
                            {
                                System.out.println("error!");
                                return -1;
                            }
                            result+=row1*row2*col2;
                            stack.push(row1);
                            stack.push(col2);
                        }
                        else 
                        {   int index;
                            index=item-'A';
                            stack.push(matrix[index][0]);
                            stack.push(matrix[index][1]);
                        }
                    }
                     }
		return result;
	}
	
}