2019 牛客暑期多校训练营 第五场 C generator 1 10进制矩阵快速幂

#include <bits/stdc++.h>
#include <cstring>
#define ll long long
using namespace std;
ll mod;
struct Mat {
	int n,m;
	ll mem[3][3];
	Mat(int n, int m) {
        this->n = n, this->m = m;
        memset(mem, 0, sizeof mem);
        return;
    }
	Mat operator*(const Mat &a)const{
        Mat c(n,a.m);
        for (int i = 1; i <= c.n; i++) {
            for (int j = 1; j <= c.m; j++) {
                for (int k = 1; k <= m; k++)
                    c.mem[i][j] += mem[i][k] * a.mem[k][j];
                c.mem[i][j] %= mod;
            }
        }
        return c;
    }
};
Mat solve(Mat a,Mat c,char *s){
	Mat b = c;
	for(int i=strlen(s)-1;i>=0;i--){
		int cur = s[i] - '0';
		while(cur--) b = b*a;
		Mat tmp = a*a;
		a = tmp*tmp;
		a = a*a*tmp;
	}
	return b;
}

char s[10000000];
int main() {
	ll x0,x1,a,b;
	ll tmp;
	Mat m1(2,2),m2(2,2);
	scanf("%lld %lld %lld %lld",&x0,&x1,&a,&b);
	scanf("%s %lld",s,&mod);
	m1.mem[1][1] = 0;
	m1.mem[1][2] = b;
	m1.mem[2][1] = 1;
	m1.mem[2][2] = a;
	m2.mem[1][1] = x0;
	m2.mem[2][1] = 0;
	m2.mem[1][2] = x1;
	m2.mem[2][2] = 0;
	m1 = solve(m1,m2,s);
	printf("%lld",m1.mem[1][1]%mod);
	return 0;
}
/*
1 1 2 2
1 8

*/