题目描述:
给定一个长度为 N 的整数数列:A1,A2,…,AN。
你要重复以下操作 K 次:
每次选择数列中最小的整数(如果最小值不止一个,选择最靠前的),将其删除,并把与它相邻的整数加上被删除的数值。
输出 K 次操作后的序列。
输入格式:
第一行包含两个整数 N 和 K。
第二行包含 N 个整数, A1,A2,A3,…,AN 。
输出格式:
输出 N−K 个整数,中间用一个空格隔开,代表 K 次操作后的序列。
## 思路:
本题需要的注意一点就是:x,y是配对出现的,你一开始往队列里面插入的是原始数据,当x的值变大的时候,队列里面y对应的那一个数值还是没有变换的值,因此需要把他拿出来之后重新放入队列,保证队列里面的数是实时更新的,还有,问题运用了,可以时内联函数运行时间加快,否则的话就会超时
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<functional>
#define x first
#define y second
using namespace std;
typedef long long L;
L n, m;
L l[100010000], r[1000010], v[1000010];//l左边,r右边,v代表该点的值
priority_queue<pair<L, L>, vector<pair<L, L>>, greater<pair<L, L>>>heap;
//定义小根堆每次取出最小
void fun(int i)//代表删除这个数以及给他旁边的数加上它自己
//模拟链表~
{
l[r[i]] = l[i];//这个数的的右边的左边=左边;
r[l[i]] = r[i];//这个数的左边的右边=右边,两步操作删咯这个点
v[l[i]] += v[i];//左边加上这个数
v[r[i]] += v[i];//右边加上这个数
}
int main()
{
cin >> n >> m;
r[0] = 1, l[n + 1] = n;//定义边界
for (int i = 1; i <= n; i++)
{
cin >> v[i];
l[i] = i - 1, r[i] = i + 1;
heap.push({ v[i],i });//插入的这个数以及他的位置
}
while (m--)//m次操作
{
pair<L, L>t = heap.top();
heap.pop();
if (t.x != v[t.y])//
//这一步操作是上述"值得一提"所提出
//因为你给x上了,它相应的y并不认识这个点
//所以应该让他们重新匹配
//简单来说:
//piar x,y绑定了,y可以识别到它加变大后的样子,但是队列
//里面仍然是一个原来的数,因此需要重新入队
{
heap.push({ v[t.y],t.y });
m++;//类似于回溯吧
}
else
fun(t.y);
}
int tt = r[0];//指针搜索
while (tt != n + 1)
{
cout << v[tt] << " ";
tt = r[tt];
}
return 0;
}