zoj 2316 Matrix Multiplication

点击打开链接

题意： 

给你一个n*n的矩阵A，然后让你 求 A与 A 的转置矩阵的积。的矩阵所有元素和。

题解：

题中给的矩阵很特殊，他的转置矩阵和本身相同。

那么他们相乘其实就是， 每一行与包括自己的所有行相乘的和。

另外 x*x+y*y+2xy =（x+y）^2;

所以把每一列所有数加起来，求平方和就是答案。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int main()
{
	long long n,m,a,b;
	long long p[205050];
	int T,ca=1;
	cin>>T;
	while(T--){
		if(ca!=1) puts("");
		ca++;
		scanf("%lld%lld",&n,&m);
		memset(p,0,sizeof(p));
		for(int i=0;i<m;i++){
			scanf("%lld%lld",&a,&b);
			if(a==b) p[a]++;
			else{
				p[a]++,p[b]++;
			}	
		}
		long long ans=0;
		for(int i=0;i<205050;i++){
			if(p[i])ans+=(long long)(p[i]*p[i]);
		}
		
		printf("%lld\n",ans);
	}
	return 0;
}