SGU196_Matrix Multiplication(求矩阵与其转置矩阵乘积后各元素的和)

题目链接：http://acm.sgu.ru/problem.php?contest=0&problem=196

题意：给定一个有N个顶点，M条边的图，N* M的矩阵A中的元素a（i，j）为1当且仅当第i个顶点是第j条边的端点，否则为0。求AT*A的各个元素之和。AT为A的转置矩阵（M＊N）。（2 <= N <= 10000 , 1 <= M <= 100000)(来自nocow)

题解：易知此题就是求各个点度数的平方和，至于证明嘛，，，还是引用神牛的证明吧：

"A的转置矩阵乘以A，就相当于每次选A的两行，然后对应项相乘再相加，就得到了结果矩阵中的一个数。每一行事实上代表一条边，当选的两条边没有公共顶点时，结果矩阵中相应的数为0，对最后的答案没有贡献。当选的两条边有一个公共顶点时，答案矩阵中会出现一个1，把这1个贡献记在这个公共顶点身上。当选的两条边其实是一条边时，答案矩阵（的对角线上）会出现一个2，把这个2个贡献给这条边的两个端点各记一个。下面考察每个顶点会有多少贡献。设它的度数为k，则选择两条不同的有公共顶点的边时它的总贡献会是k*(k-1)，仅选择一条边并给端点1个贡献时它的总贡献会是k，所以它对最终答案的贡献就是k*k。所以最终的答案就是所有点的贡献的和，也就是度数的平方和。"

代码：

import java.util.Scanner;
 
public class Solution {
 
	static int a[] = new int[11111];
	static int tmp = 0;
	static long ans = 0;
 
	public static void main(String args[]) {
		Scanner cin = new Scanner(System.in);
		int n = cin.nextInt();
		int m = cin.nextInt();
		for (int i = 1; i <= m * 2; i++) {
			tmp = cin.nextInt();
			a[tmp]++;
		}
		for (int i = 1; i <= n; i++)
			ans += (long)a[i] * a[i];
		System.out.println(ans);
	}
}