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

https://www.luogu.org/problem/show?pid=2241

题目背景

1997年普及组第一题

题目描述

有一个n*m方格的棋盘，求其方格包含多少正方形、长方形

输入输出格式

输入格式：

 

n,m因为原来数据太弱，现规定m小于等于5000，n小于等于5000（原来是100,100）

 

输出格式：

 

方格包含多少正方形、长方形

 

输入输出样例

输入样例#1：

2 3

输出样例#1：

8 10

 

#include <algorithm>
#include <cstdio>

using namespace std;

long long n,m,ans1,ans2;

int main()
{
    scanf("%lld%lld",&n,&m);
    if(n>m) swap(n,m);
    for(long long i=1;i<=n;i++) ans1+=(m-i+1)*(n-i+1);
    for(long long i=1;i<=n;i++)
      for(long long j=1;j<=m;j++)
          if(i!=j) ans2+=(n-i+1)*(m-j+1);
    printf("%lld %lld",ans1,ans2);
    return 0;
}