题意:
给定N个点,M条边的图G,构造N*M矩阵A,其中a[i][j]==1表示点i是第j条边的一个顶点,否则为0。
设AT为A的转置矩阵,即将A的行和列对换。
求AT*A的元素总和。分析:
我们可以发现在A中每一列只有两个元素u和v为1,在AT中每一行也只有两个元素u和v为1。
又AT*A时,当AT处理到第i行,A处理到第j列时,M[i][j]=∑AT[i][k]*A[k][j](M为新矩阵),我们可以知道:
M的第i行总和即为AT第i行两个元素u,v的点度数总和。
即当k=u时,∑AT[i][u]*A[u][j]=deg[u],同理∑AT[i][v]*A[v][j]=deg[v]。
因此答案即为每条边的两点度数之和。
#include <cstdio>
#include <utility>
#include <iostream>
#define u first
#define v second
using namespace std;
typedef long long LL;
const int MAXN = 10009, MAXM = 100009;
int n, m;
pair<int, int> edge[MAXM];
int deg[MAXN];
LL ans = 0;
int main()
{
scanf("%d%d", &n, &m);
for(int i = 1; i <= m; ++i)
{
int u, v;
scanf("%d%d", &u, &v);
deg[u]++, deg[v]++;
edge[i] = make_pair(u, v);
}
for(int i = 1; i <= m; ++i)
ans += deg[edge[i].u]+deg[edge[i].v];
cout << ans << endl;
return 0;
}