小米笔试 小米笔试题 0919

笔试时间:2024年09月19日 秋招

历史笔试传送门:2023秋招笔试合集

第一题

题目

小明正在整理他的玩具,他遇到了一道有趣的装箱问题:他有一个容量为N的箱子,并且有n个大小为ai的玩具。除了这n个玩具外,还有c个大小均为1的填充物,它们是小明参加各种活动的纪念品,正好可以拿来填充缝隙。他的任务是确定是否可以选其中一些玩具(填充物也包含在内)放入箱子中,恰好装满箱子,而不留下任何空隙,当然,他也可以选择全部用填充物来埴满整个箱子(如果埴充物足够多的话),也即装满一箱纪念品,小明也觉得很棒!

输入描述

第一行1个整数T,表示数据组数。对于每组数据:

第一行包合三个整数N和n和c,分别表示箱子的容量和玩具的数量以及填充物数量,

第二行包含n个整数a0,a1…an,分别表示这n个玩具的大小。1≤T≤100,1≤n≤500,1≤N,c,ai≤1000

输出描述

输出T行分别表示每组数据答案

对每组数据,输出一行,如果可以恰好装满箱子,输出YES;否则,输出 NO。

样例输入

2

10 4 1

2 3 5 7

10 1 3

6

样例输出

YES

NO

参考题解

背包dp。

C++:[此代码未进行大量数据的测试,仅供参考]

#include<iostream>
#include<vector>
using namespace std;
void solve() {
    int N, n, c;
    cin >> N >> n >> c;
    vector<int> nums(n);
    for (int i = 0; i < n; i++) {
        cin >> nums[i];
    }

    vector<vector<int>> dp(n + 1, vector<int>(N + 1));
    dp[0][0] = 1;
    for (int i = 1; i <= n; i++) {
        int x = nums[i - 1];
        for (int j = 0; j <= N; j++) {
            dp[i][j] = dp[i - 1][j];
            if (j >= x) {
                dp[i][j] |= dp[i - 1][j - x];
            }
        }
    }

    for (int i = 0; i <= N; i++) {
        if (dp[n][i] && i + c >= N) {
            cout << "YES" << "\n";
            return;
        }
    }

    cout << "NO" << "\n";
}
int main() {
    int t;
    cin >> t;
    while (t--) {
        solve();
    }
}

Java:[此代码未进行大量数据的测试,仅供参考]

import java.util.*;

public class Solution {
    public static void solve() {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int n = sc.nextInt();
        int c = sc.nextInt();
        
        int[] nums = new int[n];
        for (int i = 0; i < n; i++) {
            nums[i] = sc.nextInt();
        }
        
        boolean[][] dp = new boolean[n + 1][N + 1];
        dp[0][0] = true;
        
        for (int i = 1; i <= n; i++) {
            int x = nums[i - 1];
            for (int j = 0; j <= N; j++) {
                dp[i][j] = dp[i - 1][j];
                if (j >= x) {
                    dp[i][j] |= dp[i - 1][j - x];
                }
            }
        }

        for (int i = 0; i <= N; i++) {
            if (dp[n][i] && i + c >= N) {
                System.out.println("YES");
                return;
            }
        }
        
        System.out.println("NO");
    }
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int t = sc.nextInt();
        while (t-- > 0) {
            solve();
        }
    }
}

Python:[此代码未进行大量数据的测试,仅供参考]

def solve():
    N, n, c = map(int, input().split())
    nums = list(map(int, input().split()))

    dp = [[0] * (N + 1) for _ in range(n + 1)]
    dp[0][0] = 1

    for i in range(1, n + 1):
        x = nums[i - 1]
        for j in range(N + 1):
            dp[i]

剩余60%内容,订阅专栏后可继续查看/也可单篇购买

2024 BAT笔试合集 文章被收录于专栏

持续收录字节、腾讯、阿里、美团、美团、拼多多、华为等笔试题解,包含python、C++、Java多种语言版本,持续更新中。

全部评论

相关推荐

劝退式:我也是,小米的笔试题是我目前为止做过最难得😂
投递小米集团等公司10个岗位
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务