时间限制: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[10009], n, m, x, y; int main() { scanf ("%d %d", &n, &m); for (int i = 1; i <= m; i++) { scanf ("%d %d", &x, &y); sum[x]++, sum[y]++; } int ans = 0; for (int i = 1; i <= n; i++) ans += sum[i] * sum[i]; printf ("%d", ans); return 0; }