优先队列,链表优化

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;
  }

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

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

相关推荐
呜喵王阿尔萨斯14 分钟前
git命令解析
c++·git
点云SLAM3 小时前
PyTorch 中.backward() 详解使用
人工智能·pytorch·python·深度学习·算法·机器学习·机器人
only-qi4 小时前
146. LRU 缓存
java·算法·缓存
梁辰兴5 小时前
数据结构:排序
数据结构·算法·排序算法·c·插入排序·排序·交换排序
Murphy_lx5 小时前
Lambda表达式
开发语言·c++
yangpipi-5 小时前
C++并发编程-23. 线程间切分任务的方法
开发语言·c++
野犬寒鸦5 小时前
力扣hot100:搜索二维矩阵 II(常见误区与高效解法详解)(240)
java·数据结构·算法·leetcode·面试
菜鸟得菜6 小时前
leecode kadane算法 解决数组中子数组的最大和,以及环形数组连续子数组的最大和问题
数据结构·算法·leetcode
楼田莉子6 小时前
C++算法专题学习——分治
数据结构·c++·学习·算法·leetcode·排序算法
一支鱼7 小时前
leetcode常用解题方案总结
前端·算法·leetcode