计蒜客7.24场 F Sub Matrix Sum

最新推荐文章于 2021-10-08 22:20:59 发布

原创最新推荐文章于 2021-10-08 22:20:59 发布 · 242 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

该博客讨论了一道题目，要求找到矩阵中和大于等于给定值s的最小子矩阵，通过降维和前缀和处理，利用单调栈解决此问题。

题目大意是给出一个数s，再给出一个二维矩阵，要找到“满足矩阵和大于等于s”这个条件的最小矩阵，输出最小矩阵的元素的数目。

思路：首先要降维，对每一列处理前缀和，这样我们可以得到任意两行之间的数的总和压缩到一维数组，再对一维数组进行前缀和处理，这样我们就可以枚举所有两行之间进行二分寻找最小的宽度，也就是我们的最小矩阵，但是数组里有复数，不能满足单调性，所以要用单调栈来维护单调性，二分的对象也是单调栈内存储的下标。

//#pragma GCC optimize(3,"Ofast","inline")
#include<unordered_map>
#include<unordered_set>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<functional>
#include<cstring>
#include<string>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
#include<vector>
#include<sstream>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn= 1e5+10;
const ll mod=1e9+7;
double PI=acos(-1.0);

vector<vector<ll>> v;
ll num[maxn];
int r,c;
ll s;
vector<ll>sta;

int solve()
{
    int res=INF;
    sta.clear();
    sta.push_back(0);
    for(int i=1;i<=c;i++)
    {
        int l=0,r=sta.size()-1;
        ll tmp=num[i]-s;
        int ans=-1;
        while(l<=r)
        {
            int mid=(l+r)>>1;
            if(num[sta[mid]]<=tmp)
            {
                l=mid+1;
                ans=sta[mid];
            }
            else r=mid-1;
        }
        if(ans!=-1)
            res=min(res,i-ans);
        while(!sta.empty()&&num[sta[sta.size()-1]]>=num[i])sta.pop_back();
        sta.push_back(i);
    }
    return res;
}


int main()
{
    scanf("%d%d%lld",&r,&c,&s);
    if(r<c)
    {
        v.resize(r+1);
        for(int i=0;i<=r;i++)v[i].resize(c+1);
        for(int i=1;i<=r;i++)
            for(int j=1;j<=c;j++)
                scanf("%lld",&v[i][j]);
    }
    else
    {
        v.resize(c+1);
        for(int i=0;i<=c;i++)v[i].resize(r+1);
        for(int i=1;i<=r;i++)
            for(int j=1;j<=c;j++)
                scanf("%lld",&v[j][i]);
        swap(r,c);
    }

    for(int i=1;i<=r;i++)
        for(int j=1;j<=c;j++)
            v[i][j]+=v[i-1][j];

    int ans=INF;
    for(int l=1;l<=r;l++)
        for(int rr=l;rr<=r;rr++)
        {
            num[0]=0;
            for(int i=1;i<=c;i++)
                num[i]=v[rr][i]-v[l-1][i];
            for(int i=1;i<=c;i++)
                num[i]+=num[i-1];
            int ret=solve();
            if(ret==INF) continue;
            ans=min(ans,(rr-l+1)*ret);
        }

    printf("%d\n",ans==INF?-1:ans);

    return 0;
}