Codeforces 1323B

Source

1323B. Count Subrectangles

Solution

给出两个长度为n和m的数组，让我们把他相乘，得到一个n×m的矩阵。然后给出一个k，让我们求出矩阵里，面积为k且数字全为1的矩形有多少个。
对两个数组做前缀和，然后求出长度为x的种类有多少个，将k拆分成两个因数，设为a和b，所有的pre1[a] * pre2[b]即为所求。

Code

#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 40010;
const int inf = 0x3f3f3f3f;

int n,m,k,arr[maxn],brr[maxn],pre1[maxn],pre2[maxn];
vector<pair<int,int> > v;

int main() {
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin >> n >> m >> k;
    for(int i=0;i<n;i++) {
        cin >> arr[i];
        pre1[i] = 0;
    }
    for(int i=0;i<m;i++) {
        cin >> brr[i];
        pre2[i] = 0;
    }
    int now1 = 0,now2 = 0;
    for(int i=0;i<n;i++) {
        if(arr[i] == 0) {
            for(int i=1;i<=now1;i++) {
                pre1[i] += now1 - i + 1;
            }
            now1 = 0;
        }
        else {
            now1++;
        }
    }
    if(arr[n-1] == 1) {
        for(int i=1;i<=now1;i++) {
            pre1[i] += now1 - i + 1;
        }
    }
    for(int i=0;i<m;i++) {
        if(brr[i] == 0) {
            for(int i=1;i<=now2;i++) {
                pre2[i] += now2 - i + 1;
            }
            now2 = 0;
        }
        else {
            now2++;
        }
    }
    if(brr[m-1] == 1) {
        for(int i=1;i<=now2;i++) {
            pre2[i] += now2 - i + 1;
        }
    }
    for(int i=1;i<=sqrt(k);i++) {
        if(k % i == 0) {
            v.push_back(make_pair(i,k/i));
        }
    }
    ll ans = 0;
    for(int i=0;i<v.size();i++) {
        if(v[i].first <= n && v[i].second <= m)
            ans += pre1[v[i].first] * pre2[v[i].second];
        if(v[i].first != v[i].second && v[i].first <= m && v[i].second <= n)
            ans += pre2[v[i].first] * pre1[v[i].second];
    }
    cout << ans << endl;
}