代码设计思路:
1.在一个n * m 的方格中求正方形的个数,可以枚举正方形边长的长度 len,1 - min(m,n),不难发现此时满足条件的正方形一共有
(n - len + 1) * (m - len + 1)(由于保证了,len <= min(m,n),即保证了上式中一定不为零)。
2.在一个n*m的方格中求长方形的个数,思路与求正方形大致类似,也是枚举长方形的长和宽。譬如 2 * 3 的方格中的长方形的长和宽一共有{(1,2),(1,3),(2,3)}三种情况,即枚举宽
i( 1 <= i <= min(a,b)),
长
j (i + 1 <= j <= max(a , b)),
由于长方形有横向和纵向的两种故求长方形的个数时要考虑这两种情况:
(求个数的公式和正方形时一致的,就是个推广)
if (n - i + 1 > 0 && m - j + 1 > 0)
rec += (n - i + 1) * (m - j + 1);
if (n - j + 1 > 0 && m - i + 1 > 0)
rec += (n - j + 1) * (m - i + 1);
综上我们可以写出以下AC代码,(注意开long long 和求长方形时 i ,j 的范围,粗心的我因此wa了两发)
代码设计
#include <iostream>
#include <cstdio>
#include <numeric>
#include <functional>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <climits>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <stack>
#include <queue>
#include <bitset>
using namespace std;
typedef long long ll;
typedef pair<int, int> P;
const int maxn = 1e6 + 3;
const int mod = 1e9 + 7;
int minn = 0x3ffffff;
const int N = 1e5 + 5;
int n, m;
ll squ;
ll rec;
int main() {
cin >> n >> m;
for (int len = 1; len <= min(m, n); ++len) {
squ += (m - len + 1) * (n - len + 1);
}
for (int i = 1; i <= min(m,n); ++i) {
for (int j = i + 1; j <= max(m,n); ++j) {
if (n - i + 1 > 0 && m - j + 1 > 0)
rec += (n - i + 1) * (m - j + 1);
if (n - j + 1 > 0 && m - i + 1 > 0)
rec += (n - j + 1) * (m - i + 1);
}
}
cout << squ << ' ' << rec;
}