数方块

Time Limit: 1000 ms   Case Time Limit: 1000 ms   Memory Limit: 64 MB
Total Submission: 116   Submission Accepted: 38

Description

西瓜在寒假里辅导身为小学生的小西瓜寒假作业，里面有这样一道题目：
数一数下图中有几个正方形，小巫女认为只有16个，可是老师说错了。

对于西瓜来说，这样的问题自然很容易解决，但是现在西瓜想知道，如果现在有一个n*m个小正方形组成的长方形，请问里面一共有几个正方形呢？

Input

题目包含多组输入，EOF结束, 数据最多不超过100组，对于每组输入包含两个数字n，m表示现在有一个n*m个小正方形组成的大长方形，其中1<=n,m<=1000。

Output

对于每组输入，输出单独一行，表示大长方形中一共可以数出多少个正方形。

Sample Input

Original

Transformed

3 4
4 4
3 3 

Sample Output

Original

Transformed

20
30
14

题解
一开始用简单暴力的for循环求解，结果爆时间了
用（n*m）可以直接AC
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int main() {
	int n, m, sum;
	while (scanf("%d %d",&n,&m)!=EOF) {
		int i, j;
		sum = 0;
		int a = min(n, m), b = max(n, m);
		for (i = 1; i <= a; i++)
			for (j = 1; j <= b; j++)
				if(i==j)
				sum = sum + (b - j + 1)*(a - i + 1);
		printf("%d\n", sum);
	}
	return 0;
}