洛谷：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);
 }
}