2021CCPC华为云挑战赛 1001 对象存储调度问题

1001 对象存储调度问题

Problem Description

题目背景：在华为云的对象存储调度过程中，我们会把数据对象存入分条中，而当数据对象较小时，我们会考虑聚合多个对象到一个分条中。如图所示：

题目描述：给定 n 个大小 A1,A2,⋯,An 为 2 的整数次幂的数据对象，以及 m 个剩余空间为 B1,B2,⋯,Bm 的分条。问是否可以通过对数据对象进行聚合，使得能用现有的 m 个分条把所有 n 个对象存下来。T 组数据。

Input

第一行一个正整数 T(1≤T≤1000)，表示数据组数。

对于每组数据：

第一行两个正整数 n,m(1≤n,m≤105)，分别表示数据对象的数量和分条的数量。

第二行 n 个整数 A1,A2,⋯,An(1≤Ai≤109)，表示数据对象的大小，保证 Ai 为 2 的整数次幂。

第三行 m 个整数 B1,B2,⋯,Bm(1≤Bi≤109)，表示分条的剩余空间。

保证所有数据的 ∑n,∑m≤5×105。

Output

对于每组数据：

输出一行一个字符串 “Yes” 或者 “No”（均不含引号），分别表示分条能以及不能存下所有的数据对象。

Sample Input

2
4 2
2 2 4 8
4 12
4 2
2 2 4 8
3 13

Sample Output

Yes
No

Hint

对于第一组样例，可以把大小为 4 的数据对象放进剩余空间为 4 的分条，其余对象放入剩余空间为 12 的分条。

题目大意：

给定若干个数据和若干个空间，数据不能拆分存放，单个空间可以存放多个数据，问所有的数据能否都放入空间。

题解：

经过分析我们可以发现，如果最大的空间都放不下最大的数据，那么这些数据就不可能被全部放入空间了。所以我们可以每次都用最大的空间来放最大的数据，直到数据都放完或者最大空间放不下数据。

将所有的数据从大到小排序，每次取最大的进行存放。
同样空间也是从大到小进行排序，每次用最大的空间来存放数据。注意如果放完数据后空间有剩余，剩余的空间要重新放回空间数组中重新排序，再选出最大的空间来存放数据。

如果我们每次都对空间进行重新排序选出最大的，这样时间复杂度太高了，我们可以用一个递减的优先队列来维护空间：

priority_queue <ll, vector<ll>, less<ll> > qu; // 递减的优先队列 

这样我们可以保证队首一定是最大的空间，对于剩余的空间我们只要每次都push进队列就可以了。

剩余空间如果是0的话我们也可以放回队列，因为如果匹配到空间0的话会因为空间放不下而退出循环输出“NO”。

AC代码:

#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int N = 1e6 + 5;
ll a[N], b[N]; // a放数据，b放空间大小 
bool cmp(ll a, ll b) { // 数据从大到小排序 
    return a > b;
}
int main() {
    ios::sync_with_stdio(false);
    ll t;
    cin >> t;
    while (t --) {
        ll n, m;
        cin >> n >> m;
        priority_queue <ll, vector<ll>, less<ll> > qu; // 递减的优先队列 
        for (ll i = 0; i < n; i ++) cin >> a[i];
        for (ll i = 0; i < m; i ++) { // 空间用递减的优先队列维护 
            cin >> b[i];
            qu.push(b[i]);
        }
        sort(a, a + n, cmp); // 将数据从大到小排序 
        ll pa = 0; // 要匹配的数据下标 
        while (pa < n && qu.size() > 0) { // 如果还有数据没有匹配并且还有空间就继续比较 
            ll t = qu.top(); // 取队首最大空间 
            if (t >= a[pa]) { // 最大空间可以放下数据 
                t -= a[pa];
                qu.pop(); // 删除最大空间 
                qu.push(t); // 放回剩余空间，0也可以放回去不影响 
                pa ++; // 比较下一条数据 
            }
            else break; // 最大空间放不下数据 
        }
        if (pa >= n) cout << "Yes\n"; // 如果数据都放完了说明可以放下 
        else cout << "No\n";
    }
    return 0;
}