洛谷P1548 棋盘问题

题目描述

设有一个N×M方格的棋盘(1≤N≤100,1≤M≤100)

求出该棋盘中包含有多少个正方形、多少个长方形（不包括正方形）。

例如：当 N=2, M=3时： 

正方形的个数有8个：即边长为11的正方形有6个；

边长为2的正方形有2个。

长方形的个数有10个：

即

2×1的长方形有4个 

1×2的长方形有3个： 

3×1的长方形有2个： 

3×2的长方形有1个： 

如上例：输入：2 3

输出：8 10

输入格式

N M

输出格式

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

输入输出样例

输入 #1

2 3

输出 #1

8 10

思路

显然，固定长度n*m的矩形里面有x*y的矩形(n-（x-1）)*(m-（y-1）)个。

代码

#include <iostream>
using namespace std;

int main()
{
    int n, m;
    int square = 0, rectangle = 0;
    cin >> n >> m;
    for (int i = 1;i <= n;i++) {
        for (int j = 1;j <= m;j++) {
            if (i == j) {
                square += (n - i + 1) * (m - j + 1);
            }
            else rectangle += (n-i+1)*(m-j+1);
        }
    }
    cout << square << " " << rectangle << endl;
    return 0;
}