2247183_ac_2309ms_2860k.cpp

北大大牛代码 1240道题的原代码 超级权威

# include <stdio.h>

int n, m;
typedef struct node
{
	int	matrix[31][31];
}Mat;

Mat M;

Mat add(Mat a,Mat b)
{
	int i, j;

	for(i = 0; i < n; i++)
		for(j = 0; j < n; j++)
			a.matrix[i][j] = (a.matrix[i][j] + b.matrix[i][j])%m;
	return a;
}

Mat muti(Mat a,Mat b)
{
	Mat c;
	int i, j, k, tmp;

	for(i = 0; i < n; i++)
		for(k = 0; k < n; k++)
		{
			tmp = 0;
			for(j = 0; j < n; j++)
				tmp += a.matrix[i][j]*b.matrix[j][k], tmp%=m;
			c.matrix[i][k] = tmp;
		}
	return c;
}

Mat power(Mat a,long n)
{
	if(n==1)
		return a;
	if(n%2)
		return muti(power(a,n-1),a);
	else
		return power(muti(a,a),n/2);
}

Mat cal(long n)
{
	Mat tmp;

	if(n==1)
		return M;
	if(n%2)
		return add(cal(n-1),power(M,n));
	else
	{
		tmp = cal(n/2);
		return add(tmp,muti(tmp,power(M,n/2)));
	}
}

long input()
{
	int i, j;
	long k;

	scanf("%d%ld%d",&n,&k,&m);
	for(i = 0; i < n; i++)
		for(j = 0; j < n; j++)
			scanf("%d",&M.matrix[i][j]);
	return k;
}

void output(Mat tmp)
{
	int i, j;

	for(i = 0; i < n; i++)
		for(j = 0; j < n; j++)
		{
			printf("%d",tmp.matrix[i][j]);
			if(j==n-1)
				printf("\n");
			else
				printf(" ");
		}
}

int main()
{
	output(cal(input()));
	return 1;
}