贪心水题,没仔细看题解
题意
几个人的衣服要晾晒。挂衣夹满足如下要求:
- 一只袜子用一个衣夹
- 一件T恤用三个衣夹
- 一个人的袜子用同一种颜色
- 一个人的T恤用同一种颜色
- 两个人的衣服不能是同一种颜色
- 需要的衣夹颜色最少
给出一大堆衣夹,算出谁应该用哪种衣夹。
题解
全扔进set里面贪心。每次从待洗的衣服里优先拿拆分后的最大的,然后拿没拆分过的最大的,每次从衣夹里取能取的最小的,取不到的话就拆分,拆不了就无解。
代码
#include <iostream>
#include <set>
#include <queue>
int n, k;
std::multiset<int> l;
std::priority_queue<std::pair<int, int>> d;
int t;
int main(int argc, char *argv[]) {
std::ios::sync_with_stdio(false);
std::cin.tie(0);
std::cin >> n >> k;
for (int i = 0; i < n; i++) {
std::cin >> t;
d.push(std::make_pair(0, t*5));
}
for (int i = 0; i < k; i++) {
std::cin >> t;
l.insert(t);
}
l.insert(6000000);
int ans = 0;
while (!d.empty()) {
std::pair<int, int> x = d.top();
d.pop();
auto it = l.lower_bound(x.second);
if (*it == 6000000) {
if (x.first == 0) {
d.push(std::make_pair(1, x.second/5*3));
d.push(std::make_pair(1, x.second/5*2));
} else {
std::cout << "NIE\n";
exit(0);
}
} else {
l.erase(it);
ans++;
}
}
std::cout << ans << '\n';
}