小米笔试 小米笔试题 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多种语言版本,持续更新中。

查看5道真题和解析