hdu1757_A Simple Math Problem（新手）

我使用的是矩阵快速幂的方法写的这一道题目
可以根据题目构造一个第一行为十个输入，后面九行除了对角线下方的斜线是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;
		}
		
	}
}