洛谷:P1631 序列合并(普及+/提高,二叉堆与 ST 表)

题目:

分析:看似简单,还挺复杂的吧。

暴力+vector:

在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
int m;
int main()
{
    
    
 //priority_queue<int,vector<int>,greater<int> > q;
 cin>>m;
 int A[m];
 vector<int> v;
 for(int i=0;i<m;i++) cin>>A[i];
 int c;
 for(int i=0;i<m;i++)
 {
    
    
  cin>>c;
  for(int ii=0;ii<m;ii++) v.push_back(c+A[ii]);
 }
 sort(v.begin(),v.end());
 for(int i=0;i<m;i++) cout<<v[i]<<' ';
}

暴力+queue:

在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
int m;
int main()
{
    
    
 priority_queue<int,vector<int>,greater<int> > q;
 cin>>m;
 int A[m];
 vector<int> v;
 for(int i=0;i<m;i++) cin>>A[i];
 int c;
 for(int i=0;i<m;i++)
 {
    
    
  cin>>c;
  for(int ii=0;ii<m;ii++) q.push(c+A[ii]);
 }
 for(int i=0;i<m;i++)
 {
    
    
  cout<<q.top()<<' ';
  q.pop();
  } 
}

自己应该可以想到的啊。

原来这就是堆的应用:

在这里插入图片描述

代码:

#include<bits/stdc++.h>
using namespace std;
int m;
struct node{
    
    
 int x;
 int y;
 bool operator < (const node & nn) const
 {
    
    
  return y>nn.y;
 }
};
int main()
{
    
    
 cin>>m;
 int A[m];
 int B[m];
 priority_queue<node> q;
 for(int i=0;i<m;i++) cin>>A[i];
 for(int i=0;i<m;i++) cin>>B[i];
 for(int i=0;i<m;i++)
 {
    
    
  struct node nn;
  nn.x=0;
  nn.y=A[0]+B[i]; 
  q.push(nn);
 }
 for(int i=0;i<m;i++)
 {
    
    
  struct node nn=q.top();
  q.pop();
  cout<<nn.y<<' ';
  nn.y=nn.y-A[nn.x]+A[nn.x+1];
  nn.x++; 
  q.push(nn);
 }
}

猜你喜欢

转载自blog.csdn.net/weixin_42721412/article/details/108562959