现在的位置: 首页 > 综合 > 正文

UVa 442 – Matrix Chain Multiplication

2013年02月14日 ⁄ 综合 ⁄ 共 1184字 ⁄ 字号 评论关闭

栈题,将数据一个一个往栈里面放,直到遇到第一个‘)’ ,从‘)’元素往前找,直到第一次遇到‘(’为止,把中间出现的矩阵乘法次数算出来,然后把乘积得到的矩阵放到与‘)’对应的‘(’位置,将其覆盖,直到元素全部放完为止。

代码如下:

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<string>
using namespace std;
int rom[30]= {0},col[30]= {0},k[5]= {0},cpp;
int* multiply(char a,char b)
{
    int aa,bb;
    aa=a-'A'+1;
    bb=b-'A'+1;
    if(col[aa]!=rom[bb])
        k[3]=1;
    else
    {
        k[3]=0;
        k[0]=rom[aa]*col[bb]*col[aa];
        k[1]=rom[aa];
        k[2]=col[bb];
    }
    return k;
}
int main()
{
#ifdef test
    freopen("sample.txt","r",stdin);
#endif
    char mul[1000],c;
    int num,name[30]= {0};
    memset(name,0,sizeof(name));
    memset(mul,0,sizeof(mul));
    cin>>num;
    for(int i=1; i<=num; i++)
    {
        cin>>c;
        int kk;
        kk=c-'A'+1;
        cin>>rom[kk]>>col[kk];
    }
    cin.get();
    while(cin>>mul)
    {
        char stack[500]= {0};
        int pp=0,flag=0;cpp=0;
        int fp;
        int len=strlen(mul),sum=0;
        for(int i=0; i<len; i++)
        {
            flag=0;
            int *p;
            if(mul[i]!=')')
                stack[pp++]=mul[i];
            else
            {
                fp=pp;
                while(1)
                {
                    if(stack[fp]=='(')break;
                    fp--;
                }
                for(int i=fp+1; i<pp-1; i++)
                {
                    p=multiply(stack[i],stack[i+1]);
                    flag=p[3];
                    if(flag==1)
                        break;
                    sum+=p[0];
                    stack[fp]='A'+26+cpp;
                    rom[27+cpp]=p[1];
                    col[27+cpp]=p[2];
                }
                pp=fp+1;
                int fct=0;
                for(int ii=1; ii<pp; ii++)
                    if(stack[ii]=='(')
                        fct=1;
                if(fct==0)
                    cpp++;
            }
            if(flag==1)
                break;
        }
        if(flag==1)
            cout<<"error"<<endl;
        else
            cout<<sum<<endl;
    }
    return 0;
}

抱歉!评论已关闭.