【数论】洛谷 P1548 棋盘问题

题目描述

设有一个N*M方格的棋盘（l<=N<=100,1<=M<=100）(30％)

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

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

正方形的个数有8个：即边长为1的正方形有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

代码

#include<iostream>
using namespace std;
int main()
{
    int m,n;
    long long num1=0,num2=0;
    cin>>m>>n;
    for(int i=1;i<=n&&i<=m;i++)num1+=(m+1-i)*(n+1-i);
    num2=(1+m)*m/2*(1+n)*n/2-num1;
    cout<<num1<<' '<<num2;
    return 0;
}