CF-Round#623-div2-D题

CF-Round#623-div2-D题

D - Recommendations

传送门

这道题是一道贪心题。
用了贪心和并查集。

先开始用vis[]记录标记的时候开不了那么大。于是换了一种思路了。
经典做法应该是贪心+优先队列。

这里提供并查集把~

题目大意：有不同的报刊社，每家报刊社有一定数量的杂志（应该不是杂志，反正就是物品），然后每家报刊社都有对应自己的算法，执行一次算法需要花费t[i]的时间，可以使得该报刊社的杂志数增加1.
问最少需要多少时间可以让每家报刊社的杂志数量均不相同。

我们先直接sort下就可以啦。
原则是时间从大到小，时间相同的让数量从小到大。我们贪心策略就是改变时间小的数量大的就行。
并查集的用途：用map记录父亲结点。如果当前操作的是单个结点，我们让其父亲结点等于当前结点+1.每次这样进行。
如果我们操作到父亲结点与当前结点不同时，就说明我们碰到了数量相同的杂志数。然后现在的父结点就是最少需要加到的数目。（因为我们已经排序完毕）我们就加上其贡献，再进行父结点等于当前父结点+1的操作。方便后面继续操作。

代码部分：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;

int n;
ll ans;
struct node
{
	int num;
	int t;
}a[N]; 
map<int, int> m;

bool cmp(node a, node b)
{
	if (a.t == b.t)
	{
		return a.num < b.num;
	}
	return a.t > b.t;
}

int find(int x)
{
	return m[x] == 0 ? x : m[x] = find(m[x]);
}

void mix(int x, int y)
{
	int xx = find(x);
	int yy = find(y);
	if (xx != yy)
	{
		m[xx] = yy;
	}
}

int main()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		scanf ("%d", &a[i].num);
	}
	for (int i = 1; i <= n; i++)
	{
		scanf ("%d", &a[i].t);
	}
	sort(a + 1, a + n + 1, cmp);
	for (int i = 1; i <= n; i++)
	{
		int res = find(a[i].num);
		if (res == a[i].num)
		{
			mix(res, res + 1);
		}
		else
		{
			ans += 1ll * (res - a[i].num) * a[i].t;
			mix(res, res + 1);
		}
	}
	cout << ans << endl;
	return 0;
}