首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >2022-07-03:数组里有0和1,一定要翻转一个区间,翻转:0变1,1变0。请问翻转后可以使得1的个数最多是多少?来自小红书

2022-07-03:数组里有0和1,一定要翻转一个区间,翻转:0变1,1变0。请问翻转后可以使得1的个数最多是多少?来自小红书

作者头像
福大大架构师每日一题
发布2023-06-08 14:40:23
1720
发布2023-06-08 14:40:23
举报

2022-07-03:数组里有0和1,一定要翻转一个区间,翻转:0变1,1变0。

请问翻转后可以使得1的个数最多是多少?

来自小红书。3.13笔试。

答案2022-07-03:

某个区间,0个数-1个数=最大值。

0变成1,1变成-1,求子数组的最大累加和。

代码用rust编写。代码如下:

use rand::Rng;
fn main() {
    let nn: i32 = 100;
    let test_time: i32 = 20000;
    println!("测试开始");
    for i in 0..test_time {
        let n = rand::thread_rng().gen_range(0, nn) + 1;
        let mut arr = random_array(n);
        let mut arr2 = arr.clone();
        let ans1 = max_one_numbers1(&mut arr);
        let ans2 = max_one_numbers2(&mut arr2);
        if ans1 != ans2 {
            println!("出错了!{}", i);
            println!("ans1 = {}", ans1);
            println!("ans2 = {}", ans2);
            break;
        }
    }
    println!("测试结束");
}

fn max_one_numbers1(arr: &mut Vec<i32>) -> i32 {
    let mut ans = 0;
    for l in 0..arr.len() as i32 {
        for r in l..arr.len() as i32 {
            reverse(arr, l, r);
            ans = get_max(ans, one_numbers(arr));
            reverse(arr, l, r);
        }
    }
    return ans;
}

fn reverse(arr: &mut Vec<i32>, l: i32, r: i32) {
    for i in l..=r {
        arr[i as usize] ^= 1;
    }
}

fn one_numbers(arr: &mut Vec<i32>) -> i32 {
    let mut ans = 0;
    for num in arr.iter() {
        ans += *num;
    }
    return ans;
}

fn get_max<T: Clone + Copy + std::cmp::PartialOrd>(a: T, b: T) -> T {
    if a > b {
        a
    } else {
        b
    }
}

fn max_one_numbers2(arr: &mut Vec<i32>) -> i32 {
    let mut ans = 0;
    for num in arr.iter() {
        ans += *num;
    }
    for i in 0..arr.len() as i32 {
        arr[i as usize] = if arr[i as usize] == 0 { 1 } else { -1 };
    }
    let mut max = -2147483648;
    let mut cur = 0;
    for i in 0..arr.len() as i32 {
        cur += arr[i as usize];
        max = get_max(max, cur);
        cur = if cur < 0 { 0 } else { cur };
    }
    return ans + max;
}

// 为了测试
fn random_array(n: i32) -> Vec<i32> {
    let mut arr: Vec<i32> = vec![];
    for _i in 0..n {
        arr.push(rand::thread_rng().gen_range(0, 2));
    }
    return arr;
}

执行结果如下:

***

[左神java代码](https://github.com/algorithmzuo/weekly-problems/blob/main/src/class_2022_04_3_week/Code01_MaxOneNumbers.java)

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2022-07-03,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 福大大架构师每日一题 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
腾讯云服务器利旧
云服务器(Cloud Virtual Machine,CVM)提供安全可靠的弹性计算服务。 您可以实时扩展或缩减计算资源,适应变化的业务需求,并只需按实际使用的资源计费。使用 CVM 可以极大降低您的软硬件采购成本,简化 IT 运维工作。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档