洛谷——统计方形（数据加强版）

题目链接：统计方形（数据加强版）

思路：画图分析后，发现可以用数学公式推导出来，每列和每行的情况数目如下:
count1 = n - i + 1;
count2 = m - j + 1;
count = count1 * count2;
其中i表示当前长方形或正方形的左右边的长度，j表示当前长方形或正方形上下边的长度。当i == j时，count就是这一轮正方形的个数；否则，count就是这一轮长方形的个数。

java代码：

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		
		int n,m;
		long square_num = 0, rectangle_num = 0;
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		m = in.nextInt();
		
		for(int i=1;i<=n;++i) {
			for(int j=1;j<=m;++j) {
				long count1 = m - j + 1;
				long count2 = n - i + 1;
				if(i==j)
					square_num += count1*count2;
				else
					rectangle_num += count1*count2;
			}
		}
		
		System.out.println(square_num+ " " +rectangle_num);
	}
}