ZOJ 2316 Matrix Multiplication（思路）

题目链接

题意

给你一个 $n\ast n$ 的矩阵$A$，求 $A\ast A^T$所有元素和
给你 $m$ 条关系 $<u,v>$ 表示$A_{uv}=A_{vu}=1$

思路

由于矩阵对称
$A\ast A^T=A\ast A=E\ast A\ast A$
可以转化为图论问题，从任何一点出发，走两步到任何点的方案数。
答案即每点度数平方和

代码

#include <bits/stdc++.h>
using namespace std;

#define ll long long

ll d[10005];

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