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

矩阵乘法计算量估算

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

使用栈计算

#include <iostream>
#include <vector>
#include <stack>

using namespace std;

pair<int,int> matrix_mul(pair<int,int> a,pair<int,int> b,int& sum) {
    sum += a.first * b.second * a.second;
    
    return pair<int,int>(a.first,b.second);
}

int main() {
    
    int num;
    cin >> num;
    
    vector<pair<int,int>> matrixs;
    
    int count = num;
    while(count--) {
        int a,b;
        cin >> a >> b;
        matrixs.push_back(pair<int,int>(a,b));
    }
    
    string s;
    cin >> s;
    
    string::iterator it_start = s.begin();
    string::iterator it_end = s.end();
    
//     stack<char> s;
    stack<pair<int,int>> sp;
    
    count = 0;
    int sum=0;
    while(it_start!= it_end) {
        
        switch(*it_start) {
            case '(': 
                break;
            case ')': {
                pair<int,int> b(sp.top());
                sp.pop();
                pair<int,int> a(sp.top());
                sp.pop();
                sp.push(matrix_mul(a,b,sum));
                break;
            }
            default: 
                sp.push(matrixs[count]);
                count++;
                break;
        }
        it_start++;
    }
    
    cout << sum;
    
    return 0;
}
全部评论

相关推荐

03-26 15:18
已编辑
华北水利水电大学 Java
点赞 评论 收藏
分享
找到实习就改名4月17日下午更改:这个在牛客不是老熟人了吗
点赞 评论 收藏
分享
评论
点赞
1
分享

创作者周榜

更多
牛客网
牛客企业服务