洛谷P1801 黑匣子 双堆套路的使用
生活随笔
收集整理的這篇文章主要介紹了
洛谷P1801 黑匣子 双堆套路的使用
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題意
題目鏈接
題解
這道題本可以用Treap暴力求解出來,但是不夠優(yōu)雅,因為沒有充分利用到題目中給的條件,那就是要求的ithith小的值的ii是單調遞增的。
我們用兩個堆來維護,大頂堆和小頂堆。
大頂堆中的元素是排好序的前i?1i?1個,小頂堆里的元素個數是排好序的后n?i+1n?i+1個。
這樣每次詢問的時候,我們只需要輸出小頂堆中的最小的元素就ok了。
下面看怎么維護這兩個堆。
一開始所有的元素都往小頂堆里面插入。
當i由1變成2的時候,我們把小頂堆里的最小值扔到大頂堆里面去,始終保持大頂堆里的元素個數是i?1i?1。
如果插入一個元素比大頂堆里最大的元素還要大(或者相等),就把這個元素扔到小頂堆里面。否則扔到大頂堆里面,并且把大頂堆里面最大的元素扔到小頂堆里面去。
這就是維護過程。
代碼
#include <iostream> #include <cstdio> #include <queue> using namespace std; const int maxn = 1e6; int a[maxn],mark[maxn]; int n,m; int main(){scanf("%d%d",&m,&n);for(int i = 1;i <= m;++i)scanf("%d",&a[i]);for(int i = 0;i < n;++i){int tmp;scanf("%d",&tmp);mark[tmp]++;}priority_queue<int> Q1;priority_queue<int,vector<int>,greater<int> >Q2;int cut = 0;for(int i = 1;i <= m;++i){if(Q1.empty() || a[i] >= Q1.top())Q2.push(a[i]);else{Q1.push(a[i]);Q2.push(Q1.top());Q1.pop();}for(int j = 0;j < mark[i];++j){printf("%d\n",Q2.top());Q1.push(Q2.top());Q2.pop();}}return 0; }總結
以上是生活随笔為你收集整理的洛谷P1801 黑匣子 双堆套路的使用的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 两连跌!油价今晚迎来“大&r
- 下一篇: 洛谷 一种堆套路 P1631序列合并、P