B题 离散化+双指针

最后找到的最大区间一定是以某一个点开始的,遍历每个数,看哪个数满足,复杂度2e9

优化:

排序之后把连在一起的一块当作一个点,两个连在一块的段中间没有数的块也当作一个点,记录这些块的长度,变为5e6复杂度,最大区间一定是以某一个块的第一个数为起点开始的。双指针维护每个块为起点能到达的最远距离。

#include<bits/stdc++.h>
using namespace std;
const int M = 1000010;
const int N = 500010;
int a[N];
int idx ;
long long n,m,k;
struct Node{
    int l,r,v;
}b[M];


int main(){
    ios::sync_with_stdio();
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m>>k;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    sort(a,a+n);
    int* nn = unique(a,a+n);
    n = nn - a;
    
    b[0] = {a[0],a[0],1};
    int right = a[0];
    
    for(int i=1;i<n;i++){
        if(a[i]-1 != right){
            idx++;
            b[idx] = {right+1,a[i]-1,0};
            idx++;
            b[idx] = {a[i],a[i],1};
            right = a[i];
        }else{
            b[idx].r = a[i]; 
            right = a[i];
        }
    }
    
    long long res = 0;
    long long zero_cnt = 0;
    long long this_res = 0;
    int i,j;
    for(i=0,j=0;i<=j && i<=idx && j<=idx;i++){
        while(j <= idx){
            int j_zero=0,j_one=0;
            if(b[j].v == 1){
                j_one = b[j].r - b[j].l +1;
            }else{
                j_one = b[j].r - b[j].l +1;
                j_zero = b[j].r - b[j].l +1;
            }
            
            if(zero_cnt + j_zero >k){
                break;
            }else{
                zero_cnt += j_zero;
                this_res += j_one;
                j++;
            }
        }
        
        
        long long val_res = min(m, this_res + k - zero_cnt);
        res = max(res,val_res);
        
        int i_zero=0,i_one = 0;
        if(b[i].v == 1){
            i_one = b[i].r - b[i].l +1;
        }else{
            i_one = b[i].r - b[i].l +1;
            i_zero = b[i].r - b[i].l +1;
        }
        if(i<j){
            zero_cnt -= i_zero;
            this_res -= i_one;           
        }else{
            j=i+1;
        }
        

    }
    
    cout<<res;
}

全部评论
果然晚上脑子不好
点赞 回复 分享
发布于 2024-11-30 08:23 湖北
滑动窗口好理解 ```java for(int l = 0,r = 0; r < n;r++) {             if(r > 0) {                 k -= a[r] - a[r - 1] - 1;             }             while(k < 0) {                 k += a[l + 1] - a[l] - 1;                 l++;             }             // io.printf("left = %s\n" ,left);             ans = Math.max(ans,a[r] - a[l]  + 1 + k);         }         io.println(Math.min(ans, m)); ```
点赞 回复 分享
发布于 2024-11-30 08:22 湖北

相关推荐

来,说点可能被同行“骂”的大实话。🙊当初接数字马力Offer时,朋友都说:“蚂蚁的“内包”公司?你想清楚啊!”但入职快一年后的今天,我反而对他有了不一样的看法!🔹&nbsp;是偏见?还是信息差!之前没入职之前外面都在说什么岗位低人一等这类。实际上:这种情况不可至否,不能保证每个团队都是其乐融融。但我在的部门以及我了解的周边同事都还是十分好相处的~和蚂蚁师兄师姐之间也经常开一些小玩笑。总之:身份是蚂蚁公司给的,地位是自己挣的(一个傲娇女孩的自述)。🔹&nbsp;待遇?玩的就是真实!试用期工资全额发!六点下班跑得快(早9晚6或者早10晚7,动态打卡),公积金顶格交。别听那些画饼的,到手的钱和下班的时间才是真的(都是牛马何必难为牛马)。🔹&nbsp;能不能学到技术?来了就“后悔”!我们拥有权限直通蚂蚁知识库,技术栈多到学不完。说“学不到东西”的人,来了可能后悔——后悔来晚了(哈哈哈哈,可以不学但是不能没有)!💥&nbsp;内推地址:https://app.mokahr.com/su/ueoyhg❗我的内推码:NTA6Nvs走我的内推,可以直达业务部门,面试流程更快速,进度可查!今天新放HC,之前挂过也能再战!秋招已经正式开始啦~机会就摆在这,敢不敢来试一试呢?(和我一样,做个勇敢的女孩)
注意格局:去年超发意向是忘了
帮你内推|数字马力 校招
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务