CF 1323 B Count Subrectangles

最新推荐文章于 2024-10-09 22:28:58 发布

AMjieker

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量135

点赞数

分类专栏： CF 文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_51986723/article/details/123313937

版权

CF 专栏收录该内容

15 篇文章 0 订阅

订阅专栏

题目

在这里插入图片描述

题意

给定长为 $n$ 的数组 $a$ 和长为 $m$ 的数组 $b$ ，数组中的元素均是 $0$ 或 $1$ 。有 $n\times m$ 的矩阵 $c，c_{i, j} = a_i \times b_i$ 。请求出矩阵 $c$ 面积为 $k$ 的全 $1$ 子矩阵数量。

思路

本题数据范围比较大，考虑直接暴力是不行的。
我们可以选择一个数组，统计其中子段 $[1 - n]$ 出现的次数
对于这个子矩阵，是 $\times y$ 的，所以假设 $x$ 是层数 $a n s [k / x]$ 即可求出为x层时可以匹配的个数。
然后对于另一个数组，对于每一个1的位置，都去尽可能的匹配它前面最长段的1所构成的面积
例如 $[00100111000111]$ , $111$ 可以匹配到前3个的累加值， $11$ 可以匹配前两个的累加值。

代码

#include <bits/stdc++.h>
const int N = 2e5 + 111;
#define int long long
using namespace std;
int n, m, k;
int g[N], s[N], ans[N], base[N];
signed main() {
    cin >> n >> m >> k;
    for (int i = 1; i <= n; ++ i)
        cin >> g[i];
    for (int i = 1; i <= m; ++ i)
        cin >> s[i];
    int l = 1;
    while (l <= m) {
        while (!s[l] && l <= m) l ++;
        int cnt = 0;
        while (s[l] && l <= m) l ++, cnt ++;
        if (cnt)
            for (int i = 1; i <= cnt; ++ i)
                ans[i] += cnt - i + 1;
    }
    for (int i = 1; i <= n ; ++ i) {
        base[i] += base[i - 1];
        if (k < i) continue;
        if (k % i) continue;
        if (k / i > m) continue;
        base[i] += ans[k / i];
    }
    int pos = 0, st = 1, res = 0;
    while (st <= n) {
        if (!g[st]) pos = st;
        else res += base[st - pos];
        st ++;
    }
    cout << res << endl;
    return 0;
}