蓝桥集训之序列

阳光男孩01

已于 2024-03-09 21:33:22 修改

阅读量346

点赞数 10

文章标签：算法图论数据结构

于 2024-03-09 21:32:54 首次发布

本文链接：https://blog.csdn.net/Pisasama/article/details/136591533

版权

本文介绍了如何利用多路归并和优先队列算法解决蓝桥集训中的一个问题，通过不断合并两个已排序序列，确保每次输出的都是当前序列中最小的元素。

摘要由CSDN通过智能技术生成

蓝桥集训之序列

核心思想：多路归并
- 每次将两个序列合并 –> 两序列n²个和中最小的n个 构成新序列
- 第一行都是加b₁ 每次在最外面的元素中取最小(优先队列)

  #include<iostream>
  #include<algorithm>
  #include<cstring>
  #include<queue>
  #include<vector>
  
  using namespace std;
  const int N = 2010;
  typedef pair<int,int> PII;
  
  int a[N],b[N],c[N];
  int n,t,m;
  
  void merge()
  {
      priority_queue<PII,vector<PII>,greater<PII>> pq;
      for(int i=0;i<n;i++) pq.push({a[0] + b[i] , 0});  //将初始n个数加入优先队列
      for(int i=0;i<n;i++)
      {
          auto t = pq.top();  //取出最小
          pq.pop();
          int x = t.first , y = t.second;
          c[i] = x;
          pq.push({x - a[y] + a[y+1] , y+1});  //把后一个补上
      }
      memcpy(a, c, sizeof a);  //a[i] = c[i]更新a数组
  }
  int main()
  {
      cin>>t;
      while(t--) {
          cin >> m >> n;
          for (int i = 0; i < n; i++) scanf("%d", &a[i]);
          sort(a, a + n);
  
          for (int i = 0; i < m - 1; i++)
          {
              for(int j=0;j<n;j++)
                  scanf("%d",&b[j]);  //每输出一组 就合并
              merge();
          }
          for(int i=0;i<n;i++) printf("%d ", a[i]);
          puts(" ");
      }
  }