首页 > 代码库 > hdu 4920 Matrix multiplication(矩阵坑题)

hdu 4920 Matrix multiplication(矩阵坑题)

http://acm.hdu.edu.cn/showproblem.php?pid=4920


被这道题虐了一下午,啥也不说了。继续矩阵吧。

超时就超在每步取余上,要放在最后取余,再者注意三个循环的次序。


#include <stdio.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <cmath>
#include <stdio.h>
#include <stdlib.h>
#include <cstring>
#include <iostream>
#include <limits.h>
#include <algorithm>
#define LL long long
#define _LL __int64
using namespace std;

const int maxn = 810;

int a[maxn][maxn],b[maxn][maxn];
int ans[maxn][maxn];
int n;
int main()
{
	while(~scanf("%d",&n))
	{
		for(int i = 0; i < n; i++)
			for(int j = 0; j < n; j++)
			{
				scanf("%d",&a[i][j]);
				a[i][j] %= 3;
			}
		for(int i = 0; i < n; i++)
		{
			for(int j = 0; j < n; j++)
			{
				scanf("%d",&b[i][j]);
				b[i][j] %= 3;
			}
		}
		memset(ans,0,sizeof(ans));
		for(int i = 0; i < n; i++)
		{
			for(int k = 0; k < n; k++)
			{
				if(a[i][k] == 0) continue;
				for(int j = 0; j < n; j++)
				{
					ans[i][j] += a[i][k]*b[k][j];
				}
			}
		}

		for(int i = 0; i < n; i++)
		{
			for(int j = 0; j < n; j++)
			{
				printf("%d",ans[i][j]%3);
				if(j == n-1)
					printf("\n");
				else
					printf(" ");
			}
		}
	}
	return 0;
}