bzoj2875: [Noi2012]隨機數生成器(矩陣乘法)

Hanks_o發表於2018-03-24

題目傳送門

解法:
矩陣乘法咯。
x0 c

a 0
1 1
乘的過程中有可能爆long long。
打個類似快速冪一樣的快速乘就行。原理也是一樣。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
struct node {ll a[3][3];node(){memset(a,0,sizeof(a));}}per;ll mod;
ll ch(ll a,ll b) {
    ll ans=0;
    while(b!=0) {if(b%2ll==1ll)ans=(ans+a)%mod;a=(a+a)%mod;b/=2;}
    return ans;
}
node jc(node a,node b) {
    node c;
    for(int i=1;i<=2;i++)for(int j=1;j<=2;j++)for(int k=1;k<=2;k++)
        c.a[i][j]=(c.a[i][j]+ch(a.a[i][k],b.a[k][j]))%mod; return c;
}
node pow_mod(node a,ll b) {
    node ans=per;
    while(b!=0) {
        if(b%2==1ll) ans=jc(ans,a);
        b/=2;a=jc(a,a);
    }return ans;
}
int main() {
    node A,B;ll n,g;scanf("%lld%lld%lld%lld%lld%lld",&mod,&B.a[1][1],&A.a[1][2],&A.a[1][1],&n,&g);
    B.a[2][1]=1ll;B.a[2][2]=1ll;per.a[1][1]=1ll;per.a[2][2]=1ll;A=jc(A,pow_mod(B,n));printf("%lld\n",A.a[1][1]%g);
    return 0;
}

相關文章