优先队列,链表优化

8.整数删除 - 蓝桥云课

cpp 复制代码
#include<bits/stdc++.h> 
#define int long long
using namespace std;
typedef pair<int,int>p;
const int A=5e5+10;
int s[A];
int l[A];
int r[A];
signed main()
{
  // 请在此输入您的代码
  priority_queue<p,vector<p>,greater<p>>q;
  int N,K;
  cin>>N>>K;
  for(int i=1;i<=N;i++)
  {
    cin>>s[i];
    q.push({s[i],i});
    l[i]=i-1;
    r[i]=i+1;
    if(r[i]==N+1)
    {
      r[i]=-1;
    }
  }
  while(K)
  {
    p t=q.top();
    q.pop();
    if(t.first!=s[t.second])
    {
      q.push({s[t.second],t.second});
      continue;
    }
    K--;
    if(l[t.second]>=1)
    {
      s[l[t.second]]+=t.first;
    }
    if(r[t.second]>=1)
    {
      s[r[t.second]]+=t.first;
    }
    if(l[t.second]>=1)
    {
      r[l[t.second]]=r[t.second];
    }
    if(r[t.second]>=1)
    {
      l[r[t.second]]=l[t.second];
    }
    s[t.second]=-1;
  }
  for(int i=1;i<=N;i++)
  {
    if(s[i]!=-1)
    {
      cout<<s[i]<<" ";
    }
  }
  return 0;
}

为了找到每一次更新数组后的最小值,所以想到使用小顶堆来实现,小顶堆的定义:

cpp 复制代码
  priority_queue<p,vector<p>,greater<p>>q;

为了确定每一个数值的位置信息,从而更方便的用链表来维护这个数组,所以小顶堆是pair类型的,定义l,r数组来存储每个节点的左右坐标值,同时将超过数组范围的右节点定义为-1,方便后面的维护。

操作逻辑:循环操作次数,但这里不能简单的进行K--,因为如果取出的堆顶元素不等于堆顶元素下标值所对应的原数组元素,这是因为每一次操作时,数组中的元素的值会发生变化,所以变化后的元素值和下标需要push进堆中重新维护,这次操作不计入操作次数,所以操作次数是否减少要进行这个判断。

cpp 复制代码
    p t=q.top();
    q.pop();
    if(t.first!=s[t.second])
    {
      q.push({s[t.second],t.second});
      continue;
    }
    K--;

在进行完上面的判断后,要开始对数组进行更改:

cpp 复制代码
    if(l[t.second]>=1)
    {
      s[l[t.second]]+=t.first;
    }
    if(r[t.second]>=1)
    {
      s[r[t.second]]+=t.first;
    }
    if(l[t.second]>=1)
    {
      r[l[t.second]]=r[t.second];
    }
    if(r[t.second]>=1)
    {
      l[r[t.second]]=l[t.second];
    }
    s[t.second]=-1;
  }

对值更改:如果节点的左右节点没有越界,节点值加上堆顶值。

对下标的更改:当前节点的左节点的右节点等于当前节点的右节点,当前节点的右节点的左节点等于当前节点的左节点(删节点通用)。

相关推荐
初圣魔门首席弟子20 小时前
flag使用错误出现bug
c++·bug
Mr_WangAndy21 小时前
C++设计模式_创建型模式_原型模式Prototype
c++·设计模式·原型模式
奔跑吧邓邓子21 小时前
【C++实战㊷】C++ 原型模式实战:从概念到高效应用
c++·实战·原型模式
LeaderSheepH21 小时前
常见的排序算法
数据结构·算法·排序算法
奔跑吧邓邓子21 小时前
【C++实战㊶】C++建造者模式:复杂对象构建的秘密武器
c++·实战·建造者模式
奔跑吧邓邓子21 小时前
【C++实战㊵】C++抽象工厂模式:解锁高效对象创建的密钥
c++·实战·抽象工厂模式
jf加菲猫1 天前
条款11:优先选用删除函数,而非private未定义函数
开发语言·c++
周杰伦_Jay1 天前
【图文详解】强化学习核心框架、数学基础、分类、应用场景
人工智能·科技·算法·机器学习·计算机视觉·分类·数据挖掘
violet-lz1 天前
Linux静态库与共享库(动态库)全面详解:从创建到应用
算法
贝塔实验室1 天前
ADMM 算法的基本概念
算法·数学建模·设计模式·矩阵·动态规划·软件构建·傅立叶分析