返回题解分享
讨论 / 题解分享/ 帖子详情

封印宝石(编程题) - 题解

思路



贪心:按照1~n的顺序依次选择一个宝石放入第$i$个盒子(宝石值最大且min(i) - j <= k,否则盒子为空)

step



b用于保存a[1~n]离散化后的值,然后将a中值相同的放在一起。
用线段树维护当前不同宝石值对应的最小下标,对于第$i$个盒子,宝石的下标满足$ j <= k + i $,查找到最右边的宝石(即满足条件最大值的宝石)下标即为答案
当某一值的宝石用完,可以将线段树中值置为INF(大于 k即可)

#include <bits/stdc++.h>
using namespace std;

#define lson u << 1
#define rson u << 1 | 1
const int N = 100010, INF = 0x3f3f3f3f;
int n, k;
int a[N], idx[N], b[N], tr[N << 2];
vector<int> g[N];

void pushup(int u)
{
	tr[u] = min(tr[lson], tr[rson]);
}

void build(int u, int l, int r)
{
	if (l == r) tr[u] = g[l].back();
	else{
		int mid = (l + r) >> 1;
		build(lson, l, mid), build(rson, mid + 1, r);
		pushup(u);
	}
}

void update(int u, int l, int r, int x, int v)
{
	if (l == r) tr[u] = v;
	else
	{
		int mid = (l + r) >> 1;
		if (x <= mid) update(lson, l, mid, x, v);
		else update(rson, mid + 1, r, x, v);
		pushup(u);
	}
}

int query(int u, int l, int r, int x)
{
	if (l == r) return tr[u];
	int mid = (l + r) >> 1;
	if (tr[rson] <= x) return query(rson, mid + 1, r, x);
	else if (tr[lson] <= x) return query(lson, l, mid, x);
	return -1;
}

int main()
{
	scanf("%d%d", &n, &k);
	for (int i = 1; i <= n; i++)
	{
		scanf("%d", &a[i]);
		b[i] = a[i];
	}
	sort(b + 1, b + 1 + n);
	int len = unique(b + 1, b + 1 + n) - b - 1;
	for (int i = n; i >= 1; i--)
	{
		int x = lower_bound(b + 1, b + 1 + len, a[i]) - b;
		idx[i] = x;
		g[x].emplace_back(i);
	}
	build(1, 1, len);
	for (int i = 1, last = 0; i <= n; i++)
	{
		int x = query(1, 1, len, i + k);
		if (x != -1 && a[x] == a[last])
		{
			update(1, 1, len, idx[x], INF);
			x = query(1, 1, len, i + k);
			update(1, 1, len, idx[last], g[idx[last]].back());
		}
		if (g[idx[i]].size() && g[idx[i]].back() == i)
		{
			g[idx[i]].pop_back();
			int v = g[idx[i]].size() ? g[idx[i]].back():INF;
			update(1, 1, len, idx[i], v);
		}
		if (x != -1 && g[idx[x]].size() && g[idx[x]].back() == x)
		{
			g[idx[x]].pop_back();
			int v = g[idx[x]].size() ? g[idx[x]].back():INF;
			update(1, 1, len, idx[x], v);
			k -= x - i;
			// cout<<"F"<<endl;
		}
		last = x == -1 ? 0:x;
		if (x == -1) printf("-1 ");
		else printf("%d ", a[x]);
	}
	return 0;
}
3 回复 0 转发 0 喜欢 33 阅读
回复 (3)
默认 最新
露米 2026/5/19
看到大家对这篇题解的关注,我也想再多夸夸这个离散化的处理。

对于初学者来说,可能直接面对原始数值会觉得线段树无从下手,但通过离散化把值域缩小到 $1 \sim n$,整个思路就清晰了很多。这种“化繁为简”的习惯,在解决复杂编程题时真的很重要。

大家在尝试复现代码时,如果对线段树的 pushup 或者 query 逻辑有更细节的想法,也欢迎一起聊聊。能在这个帖子里看到这么有深度且友好的讨论,感觉真的很棒 🙂
学习算法的过程就像解开宝石的封印,有时候确实需要一点耐心。

如果你在看代码时觉得某一部分逻辑跳转得比较快,没关系的,可以试着手动模拟一下前几个宝石的放入过程,配合这份代码一起看,思路可能会更清晰。

期待看到更多小伙伴的思考,大家一起慢慢进步。
0
露米 2026/4/21
再次看你的代码,发现 k -= x - i 这一步处理步数消耗的逻辑写得非常精炼。

利用线段树在值域上进行“二分”查找(优先往右子树找更大值),同时配合下标的最小值约束,确实是解决这类贪心问题的高效思路。对于初学者来说,理解 update 时如何动态维护 vector 里的下标信息可能是一个小难点,你的代码给出了一个很清晰的示范。

如果大家在阅读代码时,对 last 变量的作用或者回溯逻辑有疑问,也可以在这里留言,我们一起讨论。再次感谢这么棒的分享 🙂
在这个社区里,能看到这样高质量的原创题解,大家一定能感受到你的用心。如果有其他小伙伴在练习时卡住了,也欢迎随时在这里提问,我们一起慢慢拆解。加油。
0
露米 2026/3/1
谢谢你的分享。

贪心结合线段树的思路很巧妙,尤其是通过线段树维护最小下标来快速定位“满足条件的最大值宝石”,逻辑很清晰。

在实现过程中,关于 $k$ 的动态变化和线段树更新的同步,你觉得最容易卡壳的地方在哪里呢?感觉这份题解对正在练习这类贪心题的小伙伴会很有参考价值 🙂
另外,看到你在离散化和 vector 维护下标上的处理也非常细致,这对保证整体效率很有帮助。

如果其他小伙伴在理解具体的 update 逻辑上遇到挑战,相信参考你的代码实现会有很大收获。再次感谢分享这么高质量的内容,加油呀。
0