题目传送门
解题思路
拿到这道题目,我由于不知道数据范围,抱着试一试的心态写了暴力代码。
暴力核心代码:
void dfs(int step, int number, int sheng) {
if (number == 0) {
ans = min(ans, step);
return;
}
for (int i = 1; i <= n; i++) {
if (!b[i]) {
b[i] = 1;
if (sheng - c[i] >= 0)
dfs(step, number - 1, sheng - c[i]);
else
dfs(step + 1, number - 1, w - c[i]);
b[i] = 0;
}
}
}
提交上去,然后只过了一个点。。。
于是我又开始想优化。如果我们现在的缆车数量大于我们已经找到的最少的数量,那么我们就不用继续找了,因为无论如何,答案都不会更新。
交上去,还是只过了一个点。。。
于是,我又开始冥思苦想,忽然想到我们可以先将所有小猫的重量进行从大到小排序,再使用一个数组 s u m sum sum 维护每辆缆车还可以承受的最大重量。假设现在已经租用 k k k 辆缆车,那么我们可以枚举 1 ∼ k 1\sim k 1∼k 辆缆车能否装下第 s t e p step step 只小猫,如果能则让这只小猫进入第 i i i 辆缆车,否则再组一辆缆车。
交上去,一片 AC
的绿色。
#include <bits/stdc++.h>
using namespace std;
int c[20], sum[20];
int n, w;
bool b[20];
int ans = 2147483646;
void dfs(int step, int number) {
if (number >= ans)
return;
if (step == n + 1) {
ans = number;
return;
}
sum[number + 1] = w - c[step];
dfs(step + 1, number + 1);
for (int i = 1; i <= number; i++) {
if (sum[i] - c[step] >= 0) {
sum[i] -= c[step];
dfs(step + 1, number);
sum[i] += c[step];
}
}
}
bool cmp(int a, int b) {
return a > b;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0), cout.tie(0);
cin >> n >> w;
for (int i = 1; i <= n; i++)
cin >> c[i];
sort(c + 1, c + 1 + n, cmp);
dfs(1, 0);
cout << ans;
return 0;
}