时间限制:0.25s
空间限制:4M
Solution
n=10000,m=100000,显然不能用矩阵乘法乘出来。
S= ATA
对于矩阵S的一行,所有在A矩阵中1位置的元素都相等,并且都等于这一行1的个数之和。假设有k个1,这一行的和显然是k*k
由此只要统计每一行有多少个1,累加它的平方就可以了。O(n)的时间解决。
code
#include<cstdio>
int sum[], n, m, x, y;
int main() {
scanf ("%d %d", &n, &m);
for (int i = ; i <= m; i++) {
scanf ("%d %d", &x, &y);
sum[x]++, sum[y]++;
}
int ans = ;
for (int i = ; i <= n; i++)
ans += sum[i] * sum[i];
printf ("%d", ans);
return ;
}