CodeForces 589G 线段树

题意：给一个数组，要求1到k的连续区间中大于d的元素与d之差值，的和大于等于r，求k的最小值，即对应的最小下标。

代码执行思路：

将不同的访问按限制d从小到大排序，则后面的元素可以继承前面的使用，节省时间。

开一个线段树，节点保存元素：sum（区间和）,min（区间最小值），max（区间最大值），cap（区间有效元素数量）。

1，当d<=min时，取sum-d*cap。2，d>=max时，删除该区间。中间多搞几次二分，并且注意对于sum的运算要用long long，即可ac，具体实现看代码。

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<iostream>

using namespace std;

const int maxn=2e5+300;

const int INF=0x3f3f3f3f;

int N,M,T,D,R,ans[maxn];

struct node

{

    int l,r,minn,maxx,del,cap;

    long long sum;

}p[maxn<<2];

struct person

{

    int d,r,id;

    bool operator <(const person &b)const

    {

        return d<b.d;

    }

}pp[maxn];

void creat_tree(int l,int r,int q)

{

    p[q].l=l,p[q].r=r,p[q].sum=0;

    p[q].del=0,p[q].minn=INF,p[q].maxx=-INF;

    p[q].cap=(r-l+1);

    if(l==r)return;

    int m=(l+r)>>1;

    creat_tree(l,m,q<<1);

    creat_tree(m+1,r,q<<1|1);

}

void dele(int q)

{

    p[q].del=1,p[q].sum=0,

    p[q].minn=INF,p[q].maxx=-INF,p[q].cap=0;

    return;

}

void push_down(int q)

{

    if(p[q].l!=p[q].r&&p[q].del==1)

    {

        dele(q<<1);

        dele(q<<1|1);

    }

    return;

}

void push_up(int q)

{

    if(p[q].l!=p[q].r)

    {

        p[q].sum=p[q<<1].sum+p[q<<1|1].sum;

        p[q].minn=min(p[q<<1].minn,p[q<<1|1].minn);

        p[q].maxx=max(p[q<<1].maxx,p[q<<1|1].maxx);

        p[q].cap=p[q<<1].cap+p[q<<1|1].cap;

        if(p[q<<1].del==1&&p[q<<1|1].del==1)dele(q);

    }

    return;

}

void insert(int val,int po,int q)

{

    if(p[q].l==p[q].r)

    {

        p[q].sum=val;

        p[q].minn=val;

        p[q].maxx=val;

        p[q].cap=1;

        return;

    }

    int m=(p[q].l+p[q].r)>>1;

    if(po<=m)

       insert(val,po,q<<1);

    else

       insert(val,po,q<<1|1);

    push_up(q);

    return;

}

long long query(int l,int r,int d,int q)

{

    push_down(q);

    if(l>p[q].r||r<p[q].l||p[q].del==1)return 0;

    if(l<=p[q].l&&r>=p[q].r&&d>=p[q].maxx)

    {

        dele(q);

        push_down(q);

        return 0;

    }

    if(l<=p[q].l&&r>=p[q].r&&d<=p[q].minn)

    {

        return p[q].sum-(long long)d*p[q].cap;

    }

    long long ta= query(l,r,d,q<<1)+query(l,r,d,q<<1|1);

    push_up(q);

    return ta;

}

int search(int di,int ri)

{

    if(query(1,M,di,1)<ri)return 0;

    int l=1,r=M+1,m,an;

    while(l<r)

    {

        m=(l+r)/2;

        if(query(1,m,di,1)>=ri)

        {

            an=m;

            r=m;

        }

        else l=m+1;

    }

    return an;

}

int main()

{

    //freopen("a.txt","r",stdin);

    //int hh=1e10,jj=1e10;

    //cout<<hh*jj<<endl;

    while(~scanf("%d%d",&N,&M))

    {

        creat_tree(1,M,1);

        for(int i=1;i<=M;i++)

        {

            scanf("%d",&T);

            insert(T,i,1);

        }

        for(int i=1;i<=N;i++)

        {

            scanf("%d%d",&D,&R);

            pp[i].d=D,pp[i].r=R,pp[i].id=i;

        }

        sort(pp+1,pp+1+N);

        for(int i=1;i<=N;i++)

        {

            ans[pp[i].id]=search(pp[i].d,pp[i].r);

        }

        for(int i=1;i<=N;i++)

        {

            printf("%d",ans[i]);

            if(i<N)printf(" ");

            else printf("\n");

        }

    }

    return 0;

}