[LOJ2274][JXOI2017]加法 二分答案+优先队列

大概思路就是:先二分答案 x ,check就是从左往右扫,如果当前值不足 x ,就贪心地不断选右端点最右的给它加上,然后用线段树\树状数组维护区间加,或者直接用优先队列维护加的次数即可。
具体地,开两个优先队列 Q,H ,前者维护当前可选区间的右端点最大值,后者维护当前已选区间左端点最小值,然后从左往右扫到 i
1. 把左端点位于 i 的区间加入 Q
2. 把左端点小于 i 的从 H 里扔掉。
3. 当前 i 位置的值应该是 Ai+sizeHa ,然后如果不足 x 就用 Q 里的区间给它加,如果 Q 不够了或者总区间超过了 k 就不合法了。
时间复杂度 O(nlog2n)
代码:

#include
#include
#include
#include
#include
#include
#define mid (L+R>>1)
#define N 200010
using namespace std;
int n,m,k,d,a[N];
vector<int> r[N];
int read()
{
    int x=0;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar());
    for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x;
}
priority_queue<int,vector<int>,greater<int> > H;
priority_queue<int> Q;
bool check(int x)
{
    while(!Q.empty()) Q.pop();
    while(!H.empty()) H.pop();
    int tk=k;
    for(int i=1;i<=n;i++)
    {   
        for(int j=0;jwhile(!H.empty()&&H.top()for(int j=a[i]+H.size()*d;jif(Q.empty()||tk<=0||Q.top()return 0;
            H.push(Q.top()),Q.pop(),tk--;
        }
    }
    return 1;
}
int main()
{
    int ca=read();
    while(ca--)
    {
        n=read();m=read();k=read();d=read();
        for(int i=1;i<=n;i++)
            a[i]=read();
        for(int i=1,x,y;i<=m;i++)
            x=read(),y=read(),r[x].push_back(y);
        int L,R;
        for(L=0,R=1e9;Lif(check(mid)) L=mid+1;
            else R=mid;
        printf("%d\n",L-1);
        for(int i=1;i<=n;i++)
            r[i].clear();
    }
    return 0;
}

你可能感兴趣的:(优先队列,二分,数据结构)