多校赛8中题

cqdp_wxy

于 2020-10-30 20:06:19 发布

阅读量296

点赞数 1

分类专栏：多校赛思维题

本文链接：https://blog.csdn.net/weixin_50773864/article/details/109391264

版权

多校赛同时被 2 个专栏收录

1 篇文章 0 订阅

订阅专栏

思维题

1 篇文章 0 订阅

订阅专栏

该博客介绍了一道编程题目，涉及矩阵运算和子矩阵元素之和的计算。题目要求根据两个序列构造矩阵，并找出所有和为给定常数的子矩阵数量。博主分享了解题思路，包括使用前缀和优化二分查找的方法，以及代码实现。最终给出了AC（Accepted）的代码示例。

摘要由CSDN通过智能技术生成

(T3)矩阵

题目背景：

14 已经想不出新颖的题目背景了，所以他准备用序列来构造矩阵。

题目描述：

WY ~~（机房巨佬）~~ 给了 14 一个长度为 $n$ 的序列 $a$ 和一个长度为 $m$ 的序列 $b$ 。

14 玩序列玩腻了，准备用这两个序列构造出一个 $n\times m$ 的矩阵 c。

14 规定 $c_{i,j}=a_i \times b_i$ ，14 是一个好奇的小朋友，

他想知道在这个矩阵中有多少个子矩阵的元素之和为 $k$ 。

其中 $k$ 为 14 给定的一个常数。

输入格式：

输入第一行包含三个整数 $n, m, k$ ，表示序列 $a$ 的长度，序列 $b$ 的长度以及 14 给出的常数。

输入第二行包含 $n$ 个整数，表示序列 $a$ 。

输入第三行包含 $m$ 个整数，表示序列 $b$ 。

输出格式：

输出只有一行，表示符合条件的子矩阵个数。

样例输入：

3 4 20
2 3 5
4 6 3 2

样例输出：

题解：

对于一个左上角为 $a_i,b_i$ ,右下角为 $a_j,b_j$ ,的矩形里面的元素和

是等于 $(\sum\limits_{k=i}^{j}a_k) \times (\sum\limits_{t=i}^{j}b_t)$

因为是 $i$ 至 $j$ 范围中间所有的 $a$ 数组里值的和

就可以先求一个前缀和

$b$ 数组同理

保证可以二分的条件：

因为所有的 $a_i,b_i$ 值都是 $1e9\ge a_i,b_i \ge0$

所以在确定左端点的情况下它是单调递增的。

把 $k$ 的所有因数求出来(不会太多，据 $l j s$ 所言最多240个左右)

预处理干完了开始主要部分：

先对 $a$ 数组枚举左端点在枚举 $k$ 的所有因数

然后对 $i$ 到 $n$ 进行二分

用之前求好的前缀和把 $a_i$ 到 $a [$ 二分的值 $]$ 的和调出来看是否等于枚举的 $k$ 的因数

$b$ 数组同理

然后就完了^_^

上代码：

#include <bits/stdc++.h>
using namespace std;
#define LL long long
LL x[1000005];
LL y[1000005];
LL vis[1000005];
LL sum[10005];
LL cnt;
LL a[50005];
LL b[50005];
void read(LL& x) {
    int f = 1;
    x = 0;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-')
            f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') {
        x *= 10;
        x += c - '0';
        c = getchar();
    }
    x *= f;
}
int main() {
    LL n, m, k;
    read(n);
    read(m);
    read(k);
    for (int i = 1; i <= k; i++) {
        if (k % i == 0) {
            sum[++cnt] = i;
        }
    }
    for (int i = 1; i <= n; i++) {
        read(a[i]);
        a[i] += a[i - 1];
    }
    for (int i = 1; i <= m; i++) {
        read(b[i]);
        b[i] += b[i - 1];
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= cnt; j++) {
            LL l = i, r = n;
            LL t = 0;
            while (l < r - 1) {
                LL mid = (l + r) / 2;
                if (a[mid] - a[i - 1] == sum[j]) {
                    x[sum[j]]++;
                    t = 1;
                    break;
                } else if (a[mid] - a[i - 1] > sum[j]) {
                    r = mid - 1;
                    continue;
                } else {
                    l = mid + 1;
                    continue;
                }
            }
            if (t == 0) {
                if (a[l] - a[i - 1] == sum[j])
                    x[sum[j]]++;
                if (a[l + 1] - a[i - 1] == sum[j])
                    x[sum[j]]++;
            }
        }
    }
    for (int i = 1; i <= m; i++) {
        for (int j = 1; j <= cnt; j++) {
            LL l = i, r = n;
            LL t = 0;
            while (l < r - 1) {
                LL mid = (l + r) / 2;
                if (b[mid] - b[i - 1] == sum[j]) {
                    t = 1;
                    y[sum[j]]++;
                    break;
                } else if (b[mid] - b[i - 1] > sum[j]) {
                    r = mid - 1;
                    continue;
                } else {
                    l = mid + 1;
                    continue;
                }
            }
            if (t == 0) {
                if (b[l] - b[i - 1] == sum[j])
                    y[sum[j]]++;
                if (b[l + 1] - b[i - 1] == sum[j])
                    y[sum[j]]++;
            }
        }
    }
    LL ans = 0;
    for (int i = 1; i <= cnt; i++) {
        ans += x[sum[i]] * y[k / sum[i]];
    }
    printf("%lld", ans);
    return 0;
}