洛谷 P3377 【模板】左偏树(可并堆)(set启发式合并)

题目描述

如题,一开始有N个小根堆,每个堆包含且仅包含一个数。接下来需要支持两种操作:

操作1: 1 x y 将第x个数和第y个数所在的小根堆合并(若第x或第y个数已经被删除或第x和第y个数在用一个堆内,则无视此操作)

操作2: 2 x 输出第x个数所在的堆最小数,并将其删除(若第x个数已经被删除,则输出-1并无视删除操作)

输入格式

第一行包含两个正整数N、M,分别表示一开始小根堆的个数和接下来操作的个数。

第二行包含N个正整数,其中第i个正整数表示第i个小根堆初始时包含且仅包含的数。

接下来M行每行2个或3个正整数,表示一条操作,格式如下:

操作1 : 1 x y

操作2 : 2 x

输出格式

输出包含若干行整数,分别依次对应每一个操作2所得的结果。

输入输出样例

输入 #1复制

5 5
1 5 4 2 3
1 1 5
1 2 5
2 2
1 4 2
2 2

输出 #1复制

1
2

说明/提示

当堆里有多个最小值时,优先删除原序列的靠前的,否则会影响后续操作1导致WA。

时空限制:1000ms,128M

数据规模:

对于30%的数据:N<=10,M<=10

对于70%的数据:N<=1000,M<=1000

对于100%的数据:N<=100000,M<=100000

样例说明:

初始状态下,五个小根堆分别为:{1}、{5}、{4}、{2}、{3}。

第一次操作,将第1个数所在的小根堆与第5个数所在的小根堆合并,故变为四个小根堆:{1,3}、{5}、{4}、{2}。

第二次操作,将第2个数所在的小根堆与第5个数所在的小根堆合并,故变为三个小根堆:{1,3,5}、{4}、{2}。

第三次操作,将第2个数所在的小根堆的最小值输出并删除,故输出1,第一个数被删除,三个小根堆为:{3,5}、{4}、{2}。

第四次操作,将第4个数所在的小根堆与第2个数所在的小根堆合并,故变为两个小根堆:{2,3,5}、{4}。

第五次操作,将第2个数所在的小根堆的最小值输出并删除,故输出2,第四个数被删除,两个小根堆为:{3,5}、{4}。

故输出依次为1、2。

我不会左偏树啊, 并查集维护加set启发式合并整一波

#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;

#ifdef LOCAL
#define debug(x) cout << "[" __FUNCTION__ ": " #x " = " << (x) << "]\n"
#define TIME cout << "RuningTime: " << clock() << "ms\n", 0
#else
#define TIME 0
#endif
#define hash_ 1000000009
#define Continue(x) { x; continue; }
#define Break(x) { x; break; }
const int mod = 998244853;
const int N = 1e6 + 10;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
ll fpow(ll a, ll b, ll mod) { ll res = 1; for (; b > 0; b >>= 1) { if (b & 1) res = res * a % mod; a = a * a % mod; } return res; }
int vis[N];
int u[N];
int pre[N];
struct node
{
	int val, idx;
	bool operator < (const node &oth)const
	{
		if (val == oth.val)
			return idx < oth.idx;
		return val < oth.val;
	}
};
set<node>s[N];
int find(int x)
{
	return pre[x] == x ? x : pre[x] = find(pre[x]);
}
int main()
{
#ifdef LOCAL
	freopen("D:/input.txt", "r", stdin);
#endif
	int n, m;
	scanf("%d%d", &n, &m);
	int num;
	for (int i = 1; i <= n; i++)
	{
		pre[i] = i;
		scanf("%d", &num);
		s[i].insert({ num, i });
	}
	for (int i = 1; i <= m; i++)
	{
		int op;
		scanf("%d", &op);
		if (op == 1)
		{
			int a, b;
			scanf("%d%d", &a, &b);
			if (vis[a] || vis[b])
				continue;
			a = find(a), b = find(b);
			if (a == b)
				continue;
			if (s[a].size() < s[b].size())
				swap(s[a], s[b]);
			for (auto &i : s[b])
				s[a].insert(i);
			pre[b] = a;
		}
		else
		{
			int num;
			scanf("%d", &num);
			if (!vis[num])
				num = find(num), printf("%d\n", s[num].begin()->val), vis[s[num].begin()->idx] = 1, s[num].erase(s[num].begin());
			else
				puts("-1");
		}
	}
	return TIME;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值