1323B. Count Subrectangles

题解

题意：

给你一个行矩阵a和一个矩阵b，都是01矩阵，问两个矩阵相乘得到的矩阵c里有多少个面积为k的全为1的矩形

官方题解：

只需要查看矩阵 $\left[\begin{array}{c}{x}_1,x_2\end{array}\right]\left[\begin{array}{c}{y}_1\\ y_2\end{array}\right]$ 里，$[x_1,x_2]$ 里有多少个 $p$，$[y_1,y_2]$ 里有多少个 $q$，满足$p\ast q=k$

---

---

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vll;
int n, m, K;
ll k;

vll cal(vi a) {
    int n = a.size();
    vll res(n + 1);//长度总共有n+1种可能
    int i = 0;
    while (i < n) {
        if (a[i] == 0) { //找到0后面第一个 a[]=1
            i++;
            continue;
        }
        int j = i;
        while (j < n && a[j]) j++;//找到1后面第一个 a[]=0
        for (int len = 1; len <= j - i; ++len) {
            res[len] += j - i - len + 1;
        }
        i = j;
    }
    return res;
}

int main() {
    ios::sync_with_stdio(0);

    cin >> n >> m >> k;
    vi a(n);
    vi b(m);
    for (int &x:a) cin >> x;
    for (int &x:b) cin >> x;
    auto p = cal(a); //统计每个p的个数
    auto q = cal(b);
    ll ans = 0;
    for (int i = 1; i < p.size(); ++i) {
        if (k % i == 0 && k / i <= m) {
            ans+=p[i]*q[k/i];
        }
    }
    cout <<ans << endl;
    return 0;
}