首页 > 试题广场 >

数组分组

[编程题]数组分组
  • 热度指数：127840 时间限制：C/C++ 1秒，其他语言2秒 空间限制：C/C++ 32M，其他语言64M
  • 算法知识视频讲解
\hspace{15pt}对于给定的 n 个整数 a_1, a_2, \dots, a_n，将其分为 a,b 两个数组，满足：
\hspace{23pt}\bullet\,所有 5 的倍数元素均在 a 数组中；
\hspace{23pt}\bullet\,所有 3 的倍数元素（不包括 5 的倍数）均在 b 数组中；
\hspace{23pt}\bullet\,其他元素可以任意分配。
\hspace{15pt}求解是否存在一种分配方案，使得 a 数组中各个元素之和等于 b 数组中各个元素之和。每一个元素要么在 a 数组中，要么在 b 数组中；数组可以为空，此时和为 0。如果存在这样的方案，输出 \rm true，否则输出 \rm false。

输入描述:
\hspace{15pt}第一行输入一个整数 n \left(1 \leqq n \leqq 30\right) 代表给定的整数个数。
\hspace{15pt}第二行输入 n 个整数 a_1, a_2, \dots, a_n \left(-500 \leqq a_i \leqq 500\right)。
\hspace{15pt}保证数据随机生成。


输出描述:
\hspace{15pt}如果存在满足条件的分配方案，输出 \rm true，否则输出 \rm false。
示例1

输入

4
1 5 -5 1

输出

true

说明

\hspace{15pt}在这个样例中，a 数组可以为 \{5, -5, 1\}，b 数组可以为 \{1\}，满足条件。
示例2

输入

3
3 5 8

输出

false
算法知识视频讲解
添加回答
Rust
牛不吃草~头像 牛不吃草~
Rust 题解，一个坑：不能用原数组的引用，必须用克隆。 
fn main() {
    let mut n = String::new();
    let mut nums = String::new();
    std::io::stdin().read_line(&mut n);
    std::io::stdin().read_line(&mut nums);

    let mut sum3 = 0;
    let mut sum5 = 0;
    let mut others = vec![];

    nums.trim()
        .split(" ")
        .map(|n| n.parse::<i32>().unwrap())
        .for_each(|n| {
            if n % 5 == 0 {
                sum5 += n;
            } else if n % 3 == 0 {
                sum3 += n;
            } else {
                others.push(n);
            }
        });

    println!("{}", backtrack(others, sum3, sum5));
}

fn backtrack(mut nums: Vec<i32>, sum1: i32, sum2: i32) -> bool {
    match nums.pop() {
        Some(x) => {
            backtrack(nums.clone(), sum1 + x, sum2) || backtrack(nums.clone(), sum1, sum2 + x)
        }
        None => sum1 == sum2,
    }
}


发表于 2022-08-22 23:33:31 回复(0)
Rust
natsunoyoru头像 natsunoyoru
一个可行的 Rust 解法，本地测试没有问题。然而牛客的用例2第二行有个莫名其妙的 '\n' 总是去不掉🤔 
use std::io;

fn is_str_numeric(s: &str) -> Option<i32> {
    for c in s.to_string().chars() {
        if !c.is_digit(10) && c != '-' {
            return None;
        }
    }
    Some(s.parse().unwrap())
}

fn have_same_sum(nums_five: &[i32], nums_three: &[i32],
                 nums_others: &[i32]) -> bool {
    if nums_others.is_empty() {
        return nums_five.iter().sum::<i32>() == nums_three.iter().sum::<i32>();
    }

    return have_same_sum(&[nums_five, &[nums_others[0]]].concat(), 
                         nums_three, 
                         &nums_others[1..])
        || have_same_sum(nums_five, 
                         &[nums_three, &[nums_others[0]]].concat(), 
                         &nums_others[1..]);
}

fn main() {
    let stdin = io::stdin();
    
    let mut n = String::new();
    while let Ok(len) = stdin.read_line(&mut n) {
        if len <= 0 {
            break;
        }
        let mut line = String::new();
        stdin.read_line(&mut line).unwrap();
        
        line.truncate(line.len() - 1);
       
        let nums: Vec<i32> = line
                                .split(" ")
                                .filter_map(|c| is_str_numeric(c))
                                .collect();
                                
        let nums_five: Vec<i32> = nums
                                        .iter()
                                        .filter(|&num| num % 5 == 0)
                                        .cloned()
                                        .collect();
        let nums_three: Vec<i32> = nums
                                        .iter()
                                        .filter(|&num| num % 3 == 0)
                                        .cloned()
                                        .collect();
        let nums_others: Vec<i32> = nums
                                        .iter()
                                        .filter(|&num| num % 3 != 0 && num % 5 != 0)
                                        .cloned()
                                        .collect();

        println!("{}", have_same_sum(nums_five.as_slice(), 
                                     nums_three.as_slice(), 
                                     nums_others.as_slice()));
    }
}


发表于 2021-12-25 20:41:13 回复(0)
提交观点

问题信息

字符串 递归
难度：
2条回答 43736浏览

热门推荐

通过挑战的用户

查看代码

相关试题

数组分组

扫一扫，把题目装进口袋

求职之前，先上牛客

扫描二维码，进入QQ群

扫描二维码，关注牛客公众号