统计方形加强版

题目背景

1997年普及组第一题

题目描述

有一个 n \times mn×m 方格的棋盘，求其方格包含多少正方形、长方形（不包含正方形）。

输入格式

一行，两个正整数 n,mn,m（n \leq 5000,m \leq 5000n≤5000,m≤5000）。

输出格式

一行，两个正整数，分别表示方格包含多少正方形、长方形（不包含正方形）。

输入输出样例

输入 #1复制

2 3

输出 #1复制

8 10

30分做法

#include<bits/stdc++.h>
using namespace std;
int main(){
	int n , m;
	long long a = 0 , b = 0;
	scanf("%d %d" , &n , &m);
	for(int x1 = 0 ; x1 < n ; x1++){
		for(int y1 = 0 ; y1 < m ; y1++){
			for(int x2 = x1 + 1 ; x2 <= n ; x2++){
				for(int y2 = y1 + 1 ; y2 <= m ; y2++){
					if(x2 - x1 == y2 - y1) a++;
					else b++;
				}
			}
		}
	}
	cout << a << " " << b;
	return 0;
} 

 满分做法时间复杂度O(nm)

#include<bits/stdc++.h>
using namespace std;
int n , m;
long long a , b , temp;
int main(){
	scanf("%d %d" , &n , &m);
	for(int i = 0 ; i <= n ; i++){
		for(int j = 0 ; j <= m ; j++){
			temp = min(i , j);
			a += temp;
			b += i * j - temp;	
		}
	}
	cout << a << " " << b;
	return 0;
}

满分做法2

枚举边长 (a, b)。题目变为在 n×m 的长方形中能放置多少 个 a×b 的方形。注意 a、b 有序，a×b 和 b×a 不等价。

• n 列中选连续 a 列：[1,a],[2,a+1],…,[n-a+1,n]，共 n-a+1 种可能。

• m 行中选连续 b 行构成方形，同理有 m-b+1 种情况。

所以 n×m 的长方形中可容纳 (n-a+1)×(m-b+1) 个 a×b 的矩形。

对于边长 k，只有长等于宽，才能构成正方形；其余均为长方形。 如果 a=b，就计入正方形。使用循环枚举 a、b，累加求和即可

//时间复杂度为O（nm） 
#include<bits/stdc++.h>
using namespace std;
int n , m;
long long a , b , temp;
int main(){
	scanf("%d %d" , &n , &m);
	for(int i = 1 ; i <= n ; i++){
		for(int j = 1 ; j <= m ; j++){
			if(i == j)
				a += (n - i + 1) * (m - j + 1);
			else
				b += (n - i + 1) * (m - j + 1);	
		}
	}
	cout << a << " " << b;
	return 0;
}

 满分做法3时间复杂度O(1)

我们是否有办法把正方形个数也算个公式出来？ 如果 n=m，枚举正方形边长，有 1×1 个 n×n 的正方形，2×2 个 (n-1)×(n-1) 的正方形，……，n×n 个 1×1 的正方形。 一共 1 2 + 2 2 + ⋯ + 𝑛 2 = 2𝑛+1 𝑛+1 𝑛 6 个正方形。 不妨设 n≤m（反之只要旋转一下就好）。一个 n×m 的长方形相 当于一个 n×n 的正方形，和剩余的 n×(m-n) 的部分。 那么考虑剩余部分。每次添加一行，这一行会带来多少的贡献？

它带来 n 个 1×1、n-1 个 2×2 … 1 个 n×n 正方形，共 (𝑛+1)*𝑛/2 个。

#include<bits/stdc++.h>
using namespace std;
long long n , m , a , b;
int main(){
	scanf("%lld %lld" , &n , &m);
	if(n > m) swap(n , m);
	a = (2 * n + 1) * (n + 1) * n / 6 + ((n + 1) * n / 2) * (m - n);
	b = n * (n + 1) * m * (m + 1) / 4 - a;
	cout << a << " " << b;
	return 0;
}