zoj 2316 Matrix Multiplication 【思路题】

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=2316；

题目大意：给 n*n的矩阵 A，求 A∗AT所有元素和（AT为转置矩阵）；
                  给m条边 <u,v> ，表示 Auv=Avu=1；

思路：A*AT=A*A*E，就相当于A中每个元素的平方，Aij 就是每个点的度数（与几个点连接）；

AC代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
#include<stack>
#include<string>
#include<math.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const double pi=acos(-1.0);
const ll mod=1e9+7;
const int N=1e5+10;

ll d[N];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(d,0,sizeof(d));
        int a,b,n,m;
        ll ans=0;
        scanf("%d %d",&n,&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d %d",&a,&b);
            d[a]++;
            d[b]++;
        }
        for(int i=1;i<=n;i++)
            ans+=d[i]*d[i];
        printf("%lld\n",ans);
        if(T) printf("\n");
    }
    return 0;
}