C. Phoenix and Towers
题意
有 n n n个石堆,有高度 h i h_i hi将它们垒成塔。现在我们要将他们合并搭建成 m m m个塔,请问能否使任意两座塔之间的高度差距小于等于 x x x。
题解
用优先队列小根堆维护。先将堆顶的元素弹出和当前读到的数据合并然后再放入优先队列即可。
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
void solve(){
int n, m, x;
cin >> n >> m >> x;
vector<int> w(n + 10);
vector<int> ans(n + 10);
priority_queue<pair<int,int>> q;
for(int i = 1;i <= m;i ++) q.push({0, i});
for(int i = 1;i <= n;i ++){
cin >> w[i];
auto [p, id] = q.top();
q.pop();
ans[i] = id;
q.push({p - w[i], id});
}
int minimum = inf, maximum = 0;
while(!q.empty()) minimum = min(minimum, -q.top().first), maximum = max(maximum, -q.top().first), q.pop();
if(maximum - minimum > x) cout << "NO" << endl;
else {
cout << "YES" << endl;
for(int i = 1;i <= n;i ++) cout << ans[i] << " ";
cout << endl;
}
return ;
}
int main(){
#ifdef DEBUG//数据量小 重定向
freopen("/Users/chenzhiyuan/Desktop/ACM/ACM/in.txt", "r", stdin);
#endif
ios_base::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int _;
cin >> _;
while(_ --) solve();
return 0;
}
D. Phoenix and Socks
题意
有 l l l只左脚袜子以及 r r r只右脚袜子,每只袜子有相同或者不同的颜色。现在有三种操作分别为更改袜子的颜色以及将左袜子换成右袜子以及将右袜子换成左袜子,每个操作进行一次都要花费一元。请问最少花费多少钱可以使所有袜子都成对。
题解
先确定以下三点
- 方向相同颜色相同修改花费 1 1 1元
- 方向不同颜色不同修改花费 1 1 1元
- 方向相同颜色不同修改花费 2 2 2元
首先可以明确的是如果存在相同的左右袜子我们可以直接配对,如果将相同的袜子都已经配对后剩余的不同的左右袜子数量相同我们可以直接将左袜子与右袜子一一配对,答案就是左袜子的数量。
如果剩余的袜子数量不同,那么我们首先将颜色相同的袜子两两配对得出将两者的数量尽量靠近,如果将所有相同颜色的袜子匹配后一者仍然比另一者大,那么我们只能将同向不同色的袜子相互匹配,每匹配一次花费 2 2 2元。
#include<bits/stdc++.h>
using namespace std;
void solve(){
int n, l, r;
cin >> n >> l >> r;
map<int, int> lmp, rmp;
for(int i = 1;i <= l;i ++){
int x;
cin >> x;
lmp[x] ++;
}
for(int i = 1;i <= r;i ++){
int x;
cin >> x;
if(lmp[x] != 0){
lmp[x] --;
}else rmp[x] ++;
}
int ans = 0;
int sum1 = 0, sum2 = 0;
for(auto v : lmp) sum1 += v.second;
for(auto v : rmp) sum2 += v.second;
if(sum1 == sum2){
cout << sum1 << endl;
return ;
}
if(sum1 > sum2){
for(auto [num, cnt] : lmp){
if(cnt >= 2){
int k = min(cnt / 2, (sum1 - sum2) / 2);
ans += k;
sum1 -= k * 2;
}
if(sum1 == sum2) break;
}
}else{
for(auto [num, cnt] : rmp){
if(cnt >= 2){
int k = min(cnt / 2, (sum2 - sum1) / 2);
ans += k;
sum2 -= k * 2;
}
if(sum1 == sum2) break;
}
}
cout << ans + (sum1 + sum2) / 2 + abs(sum1 - sum2) / 2 << endl;
return ;
}
int main(){
#ifdef DEBUG//数据量小 重定向
freopen("/Users/chenzhiyuan/Desktop/ACM/ACM/in.txt", "r", stdin);
#endif
ios_base::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int _;
cin >> _;
while(_ --) solve();
return 0;
}