我使用的是矩阵快速幂的方法写的这一道题目
可以根据题目构造一个第一行为十个输入,后面九行除了对角线下方的斜线是1,其他全为0的矩阵
如果输入 k < 10,那么直接输出k值,如果输入 k >= 10,那么输出是构造的矩阵的k - 9次幂得到的矩阵第一行与(9,8,7,6,5,4,3,2,1,0)的转置相乘后得出的值
然后上代码:
#include<iostream>
using namespace std;
long long p, q;
int* mult(int *a,int *b,int x = 10,int y = 10,int z = 10) //返回值为矩阵x行z列的a*b,其中a是x行y列,b是y行z列
{
int *ans = new int[100];
long long t;
for (int i = 0; i < 100; i++)
ans[i] = 0;
for (int i = 0; i < x; i++)
{
for (int j = 0; j < z; j++)
{
for (int k = 0; k < y; k++)
{
t = ((a[i*y + k] % q) * (b[k*z + j] % q)) % q;
ans[i*z + j] = (ans[i*z + j] + t % q) % q;
}
}
}
return ans;
}
int* pow(int *ans,int *res) //矩阵快速幂
{
while (p)
{
if (p & 1)
{
ans = mult(ans, res);
}
res = mult(res, res);
p = p >> 1;
}
return ans;
}
int main()
{
int res[100], *ans, *t;
while (scanf("%lld %d", &p, &q) != EOF)
{
/**********初始化,构造矩阵************/
ans = new int[100];
t = new int[10];
for (int i = 0; i < 10; i++)
{
t[i] = 9 - i;
for (int j = 0; j < 10; j++)
{
if (i == j)
ans[10 * i + j] = 1;
else
ans[i * 10 + j] = 0;
if (i == j + 1)
res[i * 10 + j] = 1;
else
res[i * 10 + j] = 0;
}
}
for (int i = 0; i < 10; i++)
scanf("%d", &res[i]);
/***********判断数值,决定输出************/
if (p < 10)
cout << p << endl;
else
{
int sum = 0;
p -= 9;
ans = pow(ans, res);
for (int i = 0; i < 10; i++)
{
sum = (sum + (9 - i)*ans[i] % q) % q;
}
cout << sum << endl;
}
}
}