【CF 732E】Sockets（优先队列+贪心）

【CF 732E】Sockets（优先队列+贪心）

题目大意：
n台电脑，m个供电器。
每台电脑和每个供电器都有电量，当电脑i的电量和供电器j电量相同时，可以连接供电。

现在提供变压器，可以连接到供电器上，每个变压器会将电量变为$\lceil \frac{x}{2}\rceil$变压器可以累加。提供无限个变压器。

现在问最多能使多少台电脑供上电，输出最多的可供电电脑数，最少使用的变压器数。之后输出每个电源上安装的变压器数量，以及每个电脑连接到的电源编号（0表示不连接电源）

优先队列存结构体，表示电源。
结构体中有电源编号，当前电量以及使用的变压器数量。

电量大的在堆顶，相同的变压器用的少的在堆顶。

电脑按电量从大到小排序，然后遍历一遍电脑，能按电源就安。然后就没了

代码如下：

#include <iostream>
#include <cmath>
#include <vector>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <queue>
#include <stack>
#include <list>
#include <algorithm>
#include <map>
#include <set>
#define LL long long
#define VI vector<int>
#define Pr pair<int,int>
#define fread(ch) freopen(ch,"r",stdin)
#define fwrite(ch) freopen(ch,"w",stdout)

using namespace std;
const int INF = 0x3f3f3f3f;
const int msz = 10000;
const int mod = 1e9+7;
const double eps = 1e-8;
const int maxn = 212345;

struct Stock
{
	int id,power,cnt;
	bool operator >(const struct Stock a)const
	{
		return power == a.power? cnt > a.cnt: power < a.power;
	}
} s[maxn];

struct Computer
{
	int id,power;
} com[maxn];

int ned[maxn];
int ans[maxn];
priority_queue <Stock,vector<Stock>,greater<Stock> > q;

bool cmp(Computer a,Computer b)
{
	return a.power > b.power;
}

int main()
{
	//fread("in.in");
	//fread("out.out");

	int n,m;

	scanf("%d%d",&n,&m);

	for(int i = 1; i <= n; ++i) 
	{
		scanf("%d",&com[i].power);
		com[i].id = i;
	}

	sort(com+1,com+n+1,cmp);
	for(int i = 1; i <= m; ++i)
	{
		scanf("%d",&s[i].power);
		s[i].id = i;
		s[i].cnt = 0;
		q.push(s[i]);
	}

	Stock tmp;
	int x;
	LL c,u;
	c = u = 0;

	for(int i = 1; i <= n; ++i)
	{
		while(!q.empty())
		{
			tmp = q.top();
			if(tmp.power <= com[i].power) break;
			q.pop();
			tmp.cnt++;
			x = (int)ceil(tmp.power*1.0/2);
			if(x == tmp.power) continue;
			tmp.power = x;
			q.push(tmp);
		}

		if(!q.empty())
		{
			tmp = q.top();
			if(tmp.power < com[i].power) continue;
			q.pop();
			u += tmp.cnt;
			c++;
			s[tmp.id].cnt = tmp.cnt;
			ans[com[i].id] = tmp.id;
		}
	}

	printf("%lld %lld\n",c,u);
	for(int i = 1; i <= m; ++i) printf("%d ",s[i].cnt);
	puts("");
	for(int i = 1; i <= n; ++i) printf("%d ",ans[i]);
	puts("");

	return 0;
}