Lightoj 1131 - Just Two Functions

Problem link

1. 
   
2. #include <bits/stdc++.h>
3. using namespace std;
4. #define llu long long unsigned
5. llu a1, b1,c1, a2, b2, c2, f0, f1, f2, g0, g1, g2,n, m, answer1, answer2, MOD;
6. llu cm[6][6];
7. void mul(llu a[6][6],llu b[6][6]){
8.     memset(cm, 0, sizeof(cm));
9.     for(int i = 0; i <6; i++){
10.         for(int j = 0; j <6; j++){
11.             for(int k = 0; k <6; k++){
12.                 cm[i][j]+= ((a[i][k]%MOD)*(b[k][j]%MOD))%MOD;
13.                 cm[j][j]%=MOD;
14.             }
15.         }
16.     }
17.     for(int i = 0; i <6; i++){
18.         for(int j = 0; j <6; j++){
19.             a[i][j] = cm[i][j]%MOD;
20.         }
21.     }
22. }
23. void ans(llu p[6][6], llu n){
24.     llu m[6][6] = {{a1,b1,0,0,0,c1},{1,0,0,0,0,0},{0,1,0,0,0,0}, {0,0,c2,a2,b2,0},{0,0,0,1,0,0},{0,0,0,0,1,0}};
25.     if(n == 1) return;
26.     ans(p, n/2);
27.     mul(p, p);
28.     if(n%2) mul(p, m);
29. }
30. int main(){
31.     int ts;
32.     scanf("%d", &ts);
33.     for(int p = 1; p <= ts; p++){
34.     scanf("%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld", &a1,&b1, &c1, &a2, &b2,&c2, &f0, &f1, &f2,&g0, &g1, &g2, &MOD,&m);
35.     llu sf[3] = {f0,f1,f2};
36.     llu sg[3] = {g0,g1,g2};
37.     printf("Case %d:\n", p);
38.     for(int i = 1; i <=m; i++){
39.     llu f[6][6] = {{a1,b1,0,0,0,c1},{1,0,0,0,0,0},{0,1,0,0,0,0}, {0,0,c2,a2,b2,0},{0,0,0,1,0,0},{0,0,0,0,1,0}};
40.     scanf("%lld", &n);
41.     if (n<3) answer1 = sf[n]%MOD, answer2 = sg[n]%MOD;
42.     else ans(f, n-2),answer1 = (f[0][0]*f2+f[0][1]*f1+f[0][2]*f0+f[0][3]*g2+f[0][4]*g1+f[0][5]*g0)%MOD, answer2 = (f[3][0]*f2+f[3][1]*f1+f[3][2]*f0+f[3][3]*g2+f[3][4]*g1+f[3][5]*g0)%MOD;
43.     printf("%lld %lld\n",answer1%MOD, answer2%MOD);
44.     }
45.     }
46.     return 0;
47. }