题解 | #矩阵乘法计算量估算#

矩阵乘法计算量估算

https://www.nowcoder.com/practice/15e41630514445719a942e004edc0a5b

#include <cctype>
#include <iostream>
#include <stack> 
using namespace std;
/*栈便于解决字符串中判断顺序的问题
由栈来存放字符串,可以获得计算的顺序。如(A(BC))如果来的是左括号就忽略,是字母,就把对应矩阵的行列长度压到栈中,是右括号,就弹出最上面的两个矩阵的行列来计算,计算结果压回到栈中。
要同时存储行列两个数据,先用二维数组来分别接收每一组行列,往栈中压的时候选择pair<int,int>类型即可每次压两个一组的数据
关于如何计算乘法运算次数:A(m*n)B(n*k) 相乘,A一行要乘n次,然后A的这一行要和B的k列每列都算一遍,同样的过程A要算M行,所以就是m*n*k */
int main() {
    int n;
    cin>>n;
    int a[n][2];//每行用来存一个矩阵的行列
    for(int i=0;i<n;i++){
    cin>>a[i][0]>>a[i][1];}
    string s;
    cin>>s;
    int res=0;
    stack<pair<int,int>>st;
    for(int i=0;i<s.size();i++){
        if(isalpha(s[i])){//遍历字符串,如果来的是字母,就把对应的矩阵行列压到栈中
            int t=s[i]-'A';
            st.push({a[t][0],a[t][1]});
        }
        else if(s[i]==')'){//如果来的是)就开始计算当前相邻的两个矩阵的计算次数
            pair<int,int>x=st.top();
            st.pop();
            pair<int,int>y=st.top();
            st.pop();
            if(y.second==x.first){res+=y.first*y.second*x.second;}
            st.push({y.first,x.second});
        }
    }
    cout<<res<<endl;
    
}

遇到判断字符串内部的计算顺序类的问题,都可以考虑一下栈

全部评论

相关推荐

能干的三文鱼刷了10...:公司可能有弄嵌入式需要会画pcb的需求,而且pcb能快速直观看出一个人某方面的实力。看看是否有面试资格。问你问题也能ai出来,pcb这东西能作假概率不高
点赞 评论 收藏
分享
07-21 12:41
已编辑
门头沟学院 Java
steelhead:不是你的问题,这是社会的问题。
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-23 14:13
这是聊岔撇了吗,相同的话问了两遍
吴offer选手:上下文切换这一块
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务