E. Let Them Slide 【Manthan, Codefest 19 (open for everyone, rated, Div. 1 + Div. 2) 思路+set】

题目链接

题目大意

有一个n*m的矩阵,每一行有一个长度为k的数组,数组可以再不越界的情况下在当前行滑动。问你每一列的最大和是多少

解题思路

刚开始的时候,写了一发单调队列,复杂度大概在O(n*m)左右,然后果断断超时了。
正解是:
用一个in和out去存一下,每个数在哪一列开始可以出现,在哪一列结束,这个区间这个数都可以到达,然后就是用set存一下,每一行当前列的最大值,然后用ans统计一下就好了

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<set>
#include<vector>
using namespace std;
const int N=1e6+5;
pair<int,int>p;
multiset<long long>s[N];
vector<pair<long long,int>>in[N],out[N];
long long ans[N];
int main()
{
    int n,m;
    long long x;
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        int k;
        scanf("%d",&k);
        if(k<m)
        {
            in[1].push_back(make_pair(0,i));
            out[m-k].push_back(make_pair(0,i));
            in[k+1].push_back(make_pair(0,i));
            out[m].push_back(make_pair(0,i));
        }
        for(int j=1; j<=k; j++)
        {
            scanf("%lld",&x);
            in[j].push_back(make_pair(x,i));
            out[j+m-k].push_back(make_pair(x,i));
        }
    }
    for(int i=1; i<=m; i++)
    {
        for(auto v:in[i])
        {
            int a=v.first,b=v.second;
           // printf("%d %d\n",a,b);
            if(!s[b].empty())
                ans[i]-=*s[b].rbegin();
            s[b].insert(a);
            ans[i]+=*s[b].rbegin();
        }
        if(i<m)
        {
            ans[i+1]=ans[i];
            for(auto v:out[i])
            {
                int a=v.first,b=v.second;
                ans[i+1]-=*s[b].rbegin();
                s[b].erase(s[b].find(a));
                if(!s[b].empty())
                    ans[i+1]+=*s[b].rbegin();
            }
        }
    }
    for(int i=1;i<=m;i++)
        printf("%lld ",ans[i]);
    printf("\n");
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值