洛谷 P1548 [NOIP1997 普及组] 棋盘问题【python】

题目链接
浅浅的记录一下我的洛谷刷题情况

---

---

问题重述

题目描述

输入格式

N，M

输出格式

正方形的个数与长方形的个数

输入输出样例

#样例1
输入：2 3
输出：8 10

说明/提示

无

---

大概思路

详细思路请点击这里

---

完整AC代码

n, m = map(int, input().split())
num1 = 0  # 正方形的数量
num2 = 0  # 长方形的数量
a = b = 0
for i in range(min(n, m)):  # 边长为i + 1的整形形的数量
    num1 += (n - i) * (m - i)
for i in range(1, n + 1):  # 长为i的长方形的数量
    a += i
for i in range(1, m + 1):  # 宽为i的长方形的数量
    b += i
num2 = a * b - num1
print(num1, num2)