Codeforces Global Round 14

angel for

于 2022-07-22 15:06:51 发布

阅读量93

点赞数

分类专栏： codeforces 文章标签： c++ 算法

本文链接：https://blog.csdn.net/qq_65578508/article/details/125932733

版权

codeforces 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何运用优先队列解决一个关于塔的堆叠问题。程序中，每个塔的高度不超过x，因此最高和最低的塔之间的高度差最多为x。通过维护一个最小堆，博主展示了如何在O(n log m)的时间复杂度内找到每一步的最优解，并输出堆叠顺序。代码使用C++编写，主要涉及数据结构和算法的知识。

摘要由CSDN通过智能技术生成

C题

由于每个塔最高不超过x，所以说最高的一叠和最低的一叠最大差是x，只可能小于等于x，不可能大于x

#include<bits/stdc++.h>
using namespace std;
void solve()
{
	int n,m,x;
	cin>>n>>m>>x;
	priority_queue<pair<long long,int>,vector<pair<long long,int>>,greater<pair<long long,int>>>heap; 
	for(int i=1;i<=m;i++)heap.push({0,i});
	cout<<"YES\n";
	for(int i=1;i<=n;i++){
		long long temp;
		cin>>temp;
		int num=heap.top().second;
		cout<<num<<" ";
		long long now=heap.top().first+temp;
		heap.pop();
		heap.push({now,num});
	}
	cout<<"\n";
	
}
int main()
{
	ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
	int t;
	cin>>t;
	while(t--)
	{
		solve();
	}
}