[表达式求值 矩阵乘法] ZROI 2017提高10A. Calc

本文介绍了一种使用矩阵进行状态转移的算法技巧,适用于处理字符串计算问题。通过定义三元组状态 (S,M,T),利用矩阵乘法实现了数字、加减运算的快速转移,并通过矩阵快速幂进一步提升了效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

神思路,第一次见到这种操作…
构造矩阵来进行状态的转移。
A+BC ,我们记3元组 (S,M,T) 表示当前状态,其中S=A, M=B, T=BC
这样记录,每多一个字符,转移可以表示成乘上一个矩阵。
数字 k : (S,M,T)(S,M,T10+Mk)
+ : (S,M,T)(S+T,1,0)
: (S,M,T)(S,T,0)
矩阵快速幂优化。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=100005,MOD=998244353;
struct Matrix{
    int n,m,a[5][5];
    Matrix(int x=0){ memset(a,0,sizeof(a)); a[0][0]=a[1][1]=a[2][2]=a[3][3]=x; }
    Matrix operator * (const Matrix &b)const{
        Matrix c; c.n=n; c.m=b.m;
        for(int i=0;i<=c.n-1;i++)
         for(int j=0;j<=c.m-1;j++)
          for(int k=0;k<=m-1;k++)
           (c.a[i][j]+=(LL)a[i][k]*b.a[k][j]%MOD)%=MOD;
        return c;
    }
} ans; 
int n; 
Matrix getM(char ch){
    Matrix c; c.n=c.m=4;
    if(ch=='+') c.a[0][0]=c.a[2][0]=c.a[3][1]=c.a[3][3]=1; else
    if(ch=='*') c.a[0][0]=c.a[2][1]=c.a[3][3]=1; else
    c.a[0][0]=c.a[1][1]=c.a[3][3]=1, c.a[2][2]=10, c.a[1][2]=ch-'0';
    return c; 
}
char st[20];
Matrix Pow(Matrix a,int b){
    Matrix res=1; res.n=res.m=4;
    for(;b;b>>=1,a=a*a) if(b&1) res=res*a;
    return res;
}
int main(){
    freopen("zroi10A.in","r",stdin);
    freopen("zroi10A.out","w",stdout);
    scanf("%d",&n);
    ans=0; ans.n=1; ans.m=4; ans.a[0][1]=ans.a[0][3]=1;
    for(int i=1;i<=n;i++){
        int x; scanf("%d%s",&x,st+1); int len=strlen(st+1);
        Matrix res=1; res.n=res.m=4; 
        for(int i=1;i<=len;i++) res=res*getM(st[i]);
        ans=ans*Pow(res,x);
    }
    ans=ans*getM('+');
    printf("%d\n",ans.a[0][0]);
    return 0;
};
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值